diff --git a/machine-learning/app/models/constants.py b/machine-learning/app/models/constants.py index 5e48877e99..faa1e1030d 100644 --- a/machine-learning/app/models/constants.py +++ b/machine-learning/app/models/constants.py @@ -44,6 +44,18 @@ _OPENCLIP_MODELS = { "nllb-clip-base-siglip__v1", "nllb-clip-large-siglip__mrl", "nllb-clip-large-siglip__v1", + "ViT-B-16-SigLIP2__webli", + "ViT-B-32-SigLIP2-256__webli", + "ViT-L-16-SigLIP2-256__webli", + "ViT-L-16-SigLIP2-384__webli", + "ViT-L-16-SigLIP2-512__webli", + "ViT-SO400M-14-SigLIP2-378__webli", + "ViT-SO400M-14-SigLIP2__webli", + "ViT-SO400M-16-SigLIP2-256__webli", + "ViT-SO400M-16-SigLIP2-384__webli", + "ViT-SO400M-16-SigLIP2-512__webli", + "ViT-gopt-16-SigLIP2-256__webli", + "ViT-gopt-16-SigLIP2-384__webli", } diff --git a/server/src/constants.ts b/server/src/constants.ts index 3e946578ab..6c0319fcee 100644 --- a/server/src/constants.ts +++ b/server/src/constants.ts @@ -96,6 +96,18 @@ export const CLIP_MODEL_INFO: Record = { 'ViT-SO400M-14-SigLIP-384__webli': { dimSize: 1152 }, 'nllb-clip-large-siglip__mrl': { dimSize: 1152 }, 'nllb-clip-large-siglip__v1': { dimSize: 1152 }, + 'ViT-B-16-SigLIP2__webli': { dimSize: 768 }, + 'ViT-B-32-SigLIP2-256__webli': { dimSize: 768 }, + 'ViT-L-16-SigLIP2-256__webli': { dimSize: 1024 }, + 'ViT-L-16-SigLIP2-384__webli': { dimSize: 1024 }, + 'ViT-L-16-SigLIP2-512__webli': { dimSize: 1024 }, + 'ViT-SO400M-14-SigLIP2__webli': { dimSize: 1152 }, + 'ViT-SO400M-14-SigLIP2-378__webli': { dimSize: 1152 }, + 'ViT-SO400M-16-SigLIP2-256__webli': { dimSize: 1152 }, + 'ViT-SO400M-16-SigLIP2-384__webli': { dimSize: 1152 }, + 'ViT-SO400M-16-SigLIP2-512__webli': { dimSize: 1152 }, + 'ViT-gopt-16-SigLIP2-256__webli': { dimSize: 1536 }, + 'ViT-gopt-16-SigLIP2-384__webli': { dimSize: 1536 }, }; type SharpRotationData = {