helpy4/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich synced 2025-10-17 18:19:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(ml)!: switch image classification and CLIP models to ONNX (#3809)

Browse source
This commit is contained in:
Mert 2023-08-25 00:28:51 -04:00 • committed by GitHub
parent 8211afb726
commit 165b91b068
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
14 changed files with 1617 additions and 507 deletions
Download patch file Download diff file Expand all files Collapse all files

2
machine-learning/app/models/cache.py
View file

@ -46,7 +46,7 @@ class ModelCache:
model: The requested model.
"""
key = self.cache.build_key(model_name, model_type.value)
key = f"{model_name}{model_type.value}{model_kwargs.get('mode', '')}"
async with OptimisticLock(self.cache, key) as lock:
model = await self.cache.get(key)
if model is None: