From 3e54ee5052d58d06f72a19ea05dfc0e3baa0af66 Mon Sep 17 00:00:00 2001
From: Mert <101130780+mertalev@users.noreply.github.com>
Date: Fri, 15 Dec 2023 00:34:41 -0500
Subject: [PATCH] feat(ml): add new models (#5710)

---
 .../domain/smart-info/smart-info.constant.ts   | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/server/src/domain/smart-info/smart-info.constant.ts b/server/src/domain/smart-info/smart-info.constant.ts
index 6fd863495a..052e1aee0d 100644
--- a/server/src/domain/smart-info/smart-info.constant.ts
+++ b/server/src/domain/smart-info/smart-info.constant.ts
@@ -72,9 +72,18 @@ export const CLIP_MODEL_INFO: Record<string, ModelInfo> = {
   'ViT-L-14-336__openai': {
     dimSize: 768,
   },
+  'ViT-L-14-quickgelu__dfn2b': {
+    dimSize: 768,
+  },
   'ViT-H-14__laion2b-s32b-b79k': {
     dimSize: 1024,
   },
+  'ViT-H-14-quickgelu__dfn5b': {
+    dimSize: 1024,
+  },
+  'ViT-H-14-378-quickgelu__dfn5b ': {
+    dimSize: 1024,
+  },
   'ViT-g-14__laion2b-s12b-b42k': {
     dimSize: 1024,
   },
@@ -90,6 +99,15 @@ export const CLIP_MODEL_INFO: Record<string, ModelInfo> = {
   'XLM-Roberta-Large-Vit-L-14': {
     dimSize: 768,
   },
+  'XLM-Roberta-Large-ViT-H-14__frozen_laion5b_s13b_b90k': {
+    dimSize: 1024,
+  },
+  'nllb-clip-base-siglip__v1': {
+    dimSize: 768,
+  },
+  'nllb-clip-large-siglip__v1': {
+    dimSize: 1152,
+  },
 };
 
 export function cleanModelName(modelName: string): string {