Skip to content

Commit 22de845

Browse files
authored
Prepping for final MobileCLIP weight locations (#2199)
* Prepping for final MobileCLIP weight locations * Update weight locations to coreml-projects * Update mobileclip weight locations with final apple org location
1 parent 575978b commit 22de845

File tree

2 files changed

+9
-7
lines changed

2 files changed

+9
-7
lines changed

timm/models/fastvit.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1391,21 +1391,21 @@ def _cfg(url="", **kwargs):
13911391
),
13921392

13931393
"fastvit_mci0.apple_mclip": _cfg(
1394-
#hf_hub_id='timm/',
1394+
hf_hub_id='apple/mobileclip_s0_timm',
13951395
url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_s0.pt',
13961396
crop_pct=0.95,
13971397
num_classes=512, # CLIP proj dim
13981398
mean=(0., 0., 0.), std=(1., 1., 1.)
13991399
),
14001400
"fastvit_mci1.apple_mclip": _cfg(
1401-
# hf_hub_id='timm/',
1401+
hf_hub_id='apple/mobileclip_s1_timm',
14021402
url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_s1.pt',
14031403
crop_pct=0.95,
14041404
num_classes=512, # CLIP proj dim
14051405
mean=(0., 0., 0.), std=(1., 1., 1.)
14061406
),
14071407
"fastvit_mci2.apple_mclip": _cfg(
1408-
# hf_hub_id='timm/',
1408+
hf_hub_id='apple/mobileclip_s2_timm',
14091409
url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_s2.pt',
14101410
crop_pct=0.95,
14111411
num_classes=512, # CLIP proj dim

timm/models/vision_transformer_hybrid.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -386,13 +386,15 @@ def _cfg(url='', **kwargs):
386386
'vit_base_resnet50d_224.untrained': _cfg(
387387
mean=IMAGENET_DEFAULT_MEAN, std=IMAGENET_DEFAULT_STD, first_conv='patch_embed.backbone.conv1.0'),
388388

389-
'vit_base_mci_224.apple_mclip': _cfg(
390-
url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_b.pt',
389+
'vit_base_mci_224.apple_mclip_lt': _cfg(
390+
hf_hub_id='apple/mobileclip_b_lt_timm',
391+
url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_blt.pt',
391392
num_classes=512,
392393
mean=(0., 0., 0.), std=(1., 1., 1.), first_conv='patch_embed.backbone.0.conv',
393394
),
394-
'vit_base_mci_224.apple_mclip_lt': _cfg(
395-
url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_blt.pt',
395+
'vit_base_mci_224.apple_mclip': _cfg(
396+
hf_hub_id='apple/mobileclip_b_timm',
397+
url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_b.pt',
396398
num_classes=512,
397399
mean=(0., 0., 0.), std=(1., 1., 1.), first_conv='patch_embed.backbone.0.conv',
398400
),

0 commit comments

Comments
 (0)