self-hosted/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2024-11-24 08:52:28 +02:00
Code Issues Releases Activity

feat(ml): add new models (#5710)

Browse Source
This commit is contained in:
Mert 2023-12-15 00:34:41 -05:00 committed by GitHub
parent 458257847e
commit 3e54ee5052
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
server/src/domain/smart-info/smart-info.constant.ts
View File

@ -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 {