Adding the new models to the whitelist (immich-app#5736)

cClaworns · Dec 15, 2023 · 7839be3 · 7839be3
1 parent 94e11d5
commit 7839be3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/machine-learning/app/models/constants.py b/machine-learning/app/models/constants.py
@@ -26,6 +26,9 @@
     "ViT-L-14-336__openai",
     "ViT-H-14__laion2b-s32b-b79k",
     "ViT-g-14__laion2b-s12b-b42k",
+    "ViT-L-14-quickgelu__dfn2b",
+    "ViT-H-14-quickgelu__dfn5b",
+    "ViT-H-14-378-quickgelu__dfn5b",
 }
 
 
@@ -34,6 +37,9 @@
     "XLM-Roberta-Large-Vit-B-32",
     "XLM-Roberta-Large-Vit-B-16Plus",
     "XLM-Roberta-Large-Vit-L-14",
+    "XLM-Roberta-Large-ViT-H-14__frozen_laion5b_s13b_b90k",
+    "nllb-clip-base-siglip__v1",
+    "nllb-clip-large-siglip__v1",
 }
 
 

diff --git a/server/src/domain/smart-info/smart-info.constant.ts b/server/src/domain/smart-info/smart-info.constant.ts
@@ -81,7 +81,7 @@ export const CLIP_MODEL_INFO: Record<string, ModelInfo> = {
   'ViT-H-14-quickgelu__dfn5b': {
     dimSize: 1024,
   },
-  'ViT-H-14-378-quickgelu__dfn5b ': {
+  'ViT-H-14-378-quickgelu__dfn5b': {
     dimSize: 1024,
   },
   'ViT-g-14__laion2b-s12b-b42k': {