feat(ml): add preload and fp16 settings for ocr (#23576)

2025-12-20 17:25:35 +03:00 · 2025-11-06 12:55:11 -05:00
parent 2c50f2e244
commit a4ae86ce29
6 changed files with 109 additions and 30 deletions
--- a/machine-learning/immich_ml/schemas.py
+++ b/machine-learning/immich_ml/schemas.py
@@ -46,6 +46,11 @@ class ModelSource(StrEnum):
    PADDLE = "paddle"


+class ModelPrecision(StrEnum):
+    FP16 = "FP16"
+    FP32 = "FP32"
+
+
 ModelIdentity = tuple[ModelType, ModelTask]