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 = { '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 = { '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 {