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

unnecessary model type

Browse source
This commit is contained in:
mertalev 2025-09-08 19:34:08 -04:00
parent 585d093baf
commit 49ff045df3
No known key found for this signature in database
GPG key ID: DF6ABC77AAD98C95
2 changed files with 1 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

8
machine-learning/immich_ml/models/cache.py
View file

@ -38,13 +38,7 @@ class ModelCache:
async def get(
self, model_name: str, model_type: ModelType, model_task: ModelTask, **model_kwargs: Any
) -> InferenceModel:
config_key = ""
if model_type == ModelType.OCR and model_task == ModelTask.OCR:
orientation = model_kwargs.get("orientationClassifyEnabled", True)
unwarping = model_kwargs.get("unwarpingEnabled", True)
config_key = f"_o{orientation}_u{unwarping}"
key = f"{model_name}{model_type}{model_task}{config_key}"
key = f"{model_name}{model_type}{model_task}"
async with OptimisticLock(self.cache, key) as lock:
model: InferenceModel | None = await self.cache.get(key)