CLIPA reference temppory rwightman/ models for testing

mlfoundations · Oct 17, 2023 · 72b75bd · 72b75bd
1 parent c29cc9c
commit 72b75bd
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/src/open_clip/pretrained.py b/src/open_clip/pretrained.py
@@ -381,23 +381,25 @@ def _apcfg(url='', hf_hub='', **kwargs):
         webli=_slpcfg(hf_hub='timm/ViT-SO400M-SigLIP-384/'),
     ),
 
+    # FIXME update CLIPA pretrained to final home, rwightman/ is temporary for testing
     "ViT-L-14-CLIPA": dict(
-        datacomp1b=_apcfg(),
+        datacomp1b=_apcfg(hf_hub='rwightman/ViT-L-14-CLIPA-datacomp1B/'),
     ),
     "ViT-L-14-CLIPA-336": dict(
-        datacomp1b=_apcfg(),
+        datacomp1b=_apcfg(hf_hub='rwightman/ViT-L-14-CLIPA-336-datacomp1B/'),
     ),
     "ViT-H-14-CLIPA": dict(
-        datacomp1b=_apcfg(),
+        datacomp1b=_apcfg(hf_hub='rwightman/ViT-H-14-CLIPA-datacomp1B/'),
     ),
     "ViT-H-14-CLIPA-336": dict(
-        datacomp1b=_apcfg(),
-    ),
-    "ViT-bigG-14-CLIPA": dict(
-        datacomp1b=_apcfg(),
+        laion2b=_apcfg(hf_hub='rwightman/ViT-H-14-CLIPA-336-laion2B/'),
+        datacomp1b=_apcfg(hf_hub='rwightman/ViT-H-14-CLIPA-336-datacomp1B/'),
     ),
+    # "ViT-bigG-14-CLIPA": dict(
+    #     datacomp1b=_apcfg(hf_hub='rwightman/ViT-bigG-14-CLIPA-datacomp1B/'),
+    # ),
     "ViT-bigG-14-CLIPA-336": dict(
-        datacomp1b=_apcfg(),
+        datacomp1b=_apcfg(hf_hub='rwightman/ViT-bigG-14-CLIPA-336-datacomp1B/'),
     ),
 
     "nllb-clip-base": dict(