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

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( async def get(
self, model_name: str, model_type: ModelType, model_task: ModelTask, **model_kwargs: Any self, model_name: str, model_type: ModelType, model_task: ModelTask, **model_kwargs: Any
) -> InferenceModel: ) -> InferenceModel:
config_key = "" key = f"{model_name}{model_type}{model_task}"
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}"
async with OptimisticLock(self.cache, key) as lock: async with OptimisticLock(self.cache, key) as lock:
model: InferenceModel | None = await self.cache.get(key) model: InferenceModel | None = await self.cache.get(key)

1
machine-learning/immich_ml/schemas.py
View file

@ -31,7 +31,6 @@ class ModelType(StrEnum):
RECOGNITION = "recognition" RECOGNITION = "recognition"
TEXTUAL = "textual" TEXTUAL = "textual"
VISUAL = "visual" VISUAL = "visual"
OCR = "ocr"
class ModelFormat(StrEnum): class ModelFormat(StrEnum):