Added links to pretrained models.

mlfoundations · Oct 11, 2023 · 4f717ec · 4f717ec
1 parent 902f816
commit 4f717ec
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/open_clip/pretrained.py b/src/open_clip/pretrained.py
@@ -288,6 +288,12 @@ def _pcfg(url='', hf_hub='', mean=None, std=None):
     "EVA02-E-14-plus": dict(
         # from QuanSun/EVA-CLIP/EVA02_CLIP_E_psz14_plus_s9B.pt
         laion2b_s9b_b144k=_pcfg(hf_hub='timm/eva02_enormous_patch14_plus_clip_224.laion2b_s9b_b144k/'),
+    ),
+    "nllb-clip-base": dict(
+        v1=_pcfg(hf_hub='visheratin/nllb-clip-base-oc/'),
+    ),
+    "nllb-clip-large": dict(
+        v1=_pcfg(hf_hub='visheratin/nllb-clip-large-oc/'),
     )
 }