c-l/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2024-12-29 15:11:58 +00:00
Code Releases Activity

Adding the new models to the whitelist (#5736)

Browse source
This commit is contained in:
Mohamed BOUSSAID 2023-12-15 23:45:14 +01:00 committed by GitHub
parent 94e11d52dc
commit 7839be3b49
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
machine-learning/app/models/constants.py
View file

@ -26,6 +26,9 @@ _OPENCLIP_MODELS = {
"ViT-L-14-336__openai", "ViT-L-14-336__openai",
"ViT-H-14__laion2b-s32b-b79k", "ViT-H-14__laion2b-s32b-b79k",
"ViT-g-14__laion2b-s12b-b42k", "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 @@ _MCLIP_MODELS = {
"XLM-Roberta-Large-Vit-B-32", "XLM-Roberta-Large-Vit-B-32",
"XLM-Roberta-Large-Vit-B-16Plus", "XLM-Roberta-Large-Vit-B-16Plus",
"XLM-Roberta-Large-Vit-L-14", "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",
} }

2
server/src/domain/smart-info/smart-info.constant.ts
View file

@ -81,7 +81,7 @@ export const CLIP_MODEL_INFO: Record<string, ModelInfo> = {
'ViT-H-14-quickgelu__dfn5b': { 'ViT-H-14-quickgelu__dfn5b': {
dimSize: 1024, dimSize: 1024,
}, },
'ViT-H-14-378-quickgelu__dfn5b ': { 'ViT-H-14-378-quickgelu__dfn5b': {
dimSize: 1024, dimSize: 1024,
}, },
'ViT-g-14__laion2b-s12b-b42k': { 'ViT-g-14__laion2b-s12b-b42k': {