github-mirrors/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2026-05-18 03:10:24 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix(ml): batch size setting (#26524)

Browse Source
This commit is contained in:
Mert
2026-03-05 12:01:47 -05:00
committed by GitHub
parent 09fabb36b6
commit 35a521c6ec
6 changed files with 89 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
machine-learning/immich_ml/config.py
+4 -1
View File
@@ -48,8 +48,11 @@ class PreloadModelData(BaseModel):
class MaxBatchSize(BaseModel):
ocr_fallback: str | None = os.getenv("MACHINE_LEARNING_MAX_BATCH_SIZE__TEXT_RECOGNITION", None)
if ocr_fallback is not None:
os.environ["MACHINE_LEARNING_MAX_BATCH_SIZE__OCR"] = ocr_fallback
facial_recognition: int | None = None
text_recognition: int | None = None
ocr: int | None = None
class Settings(BaseSettings):