Update weight locations to coreml-projects

huggingface · Jun 12, 2024 · e91a135 · e91a135
1 parent 3c46320
commit e91a135
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/timm/models/fastvit.py b/timm/models/fastvit.py
@@ -1391,22 +1391,22 @@ def _cfg(url="", **kwargs):
     ),
 
     "fastvit_mci0.apple_mclip": _cfg(
-        hf_hub_id='reach-vb/mobileclip_s0_timm',
-        #url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_s0.pt',
+        hf_hub_id='coreml-projects/mobileclip_s0_timm',
+        url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_s0.pt',
         crop_pct=0.95,
         num_classes=512,  # CLIP proj dim
         mean=(0., 0., 0.), std=(1., 1., 1.)
     ),
     "fastvit_mci1.apple_mclip": _cfg(
-        hf_hub_id='reach-vb/mobileclip_s1_timm',
-        #url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_s1.pt',
+        hf_hub_id='coreml-projects/mobileclip_s1_timm',
+        url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_s1.pt',
         crop_pct=0.95,
         num_classes=512,  # CLIP proj dim
         mean=(0., 0., 0.), std=(1., 1., 1.)
     ),
     "fastvit_mci2.apple_mclip": _cfg(
-        hf_hub_id='reach-vb/mobileclip_s2_timm',
-        #url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_s2.pt',
+        hf_hub_id='coreml-projects/mobileclip_s2_timm',
+        url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_s2.pt',
         crop_pct=0.95,
         num_classes=512,  # CLIP proj dim
         mean=(0., 0., 0.), std=(1., 1., 1.)

diff --git a/timm/models/vision_transformer_hybrid.py b/timm/models/vision_transformer_hybrid.py
@@ -386,15 +386,15 @@ def _cfg(url='', **kwargs):
     'vit_base_resnet50d_224.untrained': _cfg(
         mean=IMAGENET_DEFAULT_MEAN, std=IMAGENET_DEFAULT_STD, first_conv='patch_embed.backbone.conv1.0'),
 
-    'vit_base_mci_224.apple_mclip': _cfg(
-        hf_hub_id='reach-vb/mobileclip_B_timm',
-        #url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_b.pt',
+    'vit_base_mci_224.apple_mclip_lt': _cfg(
+        hf_hub_id='coreml-projects/mobileclip_b_lt_timm',
+        url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_blt.pt',
         num_classes=512,
         mean=(0., 0., 0.), std=(1., 1., 1.), first_conv='patch_embed.backbone.0.conv',
     ),
-    'vit_base_mci_224.apple_mclip_lt': _cfg(
-        hf_hub_id='reach-vb/mobileclip_B_LT_timm',
-        #url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_blt.pt',
+    'vit_base_mci_224.apple_mclip': _cfg(
+        hf_hub_id='coreml-projects/mobileclip_b_timm',
+        url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_b.pt',
         num_classes=512,
         mean=(0., 0., 0.), std=(1., 1., 1.), first_conv='patch_embed.backbone.0.conv',
     ),