Add missing hf hub entries for mvitv2

huggingface · Nov 27, 2023 · cd8d9d9 · cd8d9d9
1 parent 19a8c18
commit cd8d9d9
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/timm/models/mvitv2.py b/timm/models/mvitv2.py
@@ -975,20 +975,28 @@ def _cfg(url='', **kwargs):
 
 
 default_cfgs = generate_default_cfgs({
-    'mvitv2_tiny.fb_in1k': _cfg(url='https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_T_in1k.pyth'),
-    'mvitv2_small.fb_in1k': _cfg(url='https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_S_in1k.pyth'),
-    'mvitv2_base.fb_in1k': _cfg(url='https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_B_in1k.pyth'),
-    'mvitv2_large.fb_in1k': _cfg(url='https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_L_in1k.pyth'),
+    'mvitv2_tiny.fb_in1k': _cfg(
+        url='https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_T_in1k.pyth',
+        hf_hub_id='timm/'),
+    'mvitv2_small.fb_in1k': _cfg(url='https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_S_in1k.pyth',
+        hf_hub_id='timm/'),
+    'mvitv2_base.fb_in1k': _cfg(url='https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_B_in1k.pyth',
+        hf_hub_id='timm/'),
+    'mvitv2_large.fb_in1k': _cfg(url='https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_L_in1k.pyth',
+        hf_hub_id='timm/'),
 
     'mvitv2_small_cls': _cfg(url=''),
     'mvitv2_base_cls.fb_inw21k': _cfg(
         url='https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_B_in21k.pyth',
+        hf_hub_id='timm/',
         num_classes=19168),
     'mvitv2_large_cls.fb_inw21k': _cfg(
         url='https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_L_in21k.pyth',
+        hf_hub_id='timm/',
         num_classes=19168),
     'mvitv2_huge_cls.fb_inw21k': _cfg(
         url='https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_H_in21k.pyth',
+        hf_hub_id='timm/',
         num_classes=19168),
 })