@@ -975,20 +975,28 @@ def _cfg(url='', **kwargs):
975
975
976
976
977
977
default_cfgs = generate_default_cfgs ({
978
- 'mvitv2_tiny.fb_in1k' : _cfg (url = 'https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_T_in1k.pyth' ),
979
- 'mvitv2_small.fb_in1k' : _cfg (url = 'https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_S_in1k.pyth' ),
980
- 'mvitv2_base.fb_in1k' : _cfg (url = 'https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_B_in1k.pyth' ),
981
- 'mvitv2_large.fb_in1k' : _cfg (url = 'https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_L_in1k.pyth' ),
978
+ 'mvitv2_tiny.fb_in1k' : _cfg (
979
+ url = 'https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_T_in1k.pyth' ,
980
+ hf_hub_id = 'timm/' ),
981
+ 'mvitv2_small.fb_in1k' : _cfg (url = 'https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_S_in1k.pyth' ,
982
+ hf_hub_id = 'timm/' ),
983
+ 'mvitv2_base.fb_in1k' : _cfg (url = 'https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_B_in1k.pyth' ,
984
+ hf_hub_id = 'timm/' ),
985
+ 'mvitv2_large.fb_in1k' : _cfg (url = 'https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_L_in1k.pyth' ,
986
+ hf_hub_id = 'timm/' ),
982
987
983
988
'mvitv2_small_cls' : _cfg (url = '' ),
984
989
'mvitv2_base_cls.fb_inw21k' : _cfg (
985
990
url = 'https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_B_in21k.pyth' ,
991
+ hf_hub_id = 'timm/' ,
986
992
num_classes = 19168 ),
987
993
'mvitv2_large_cls.fb_inw21k' : _cfg (
988
994
url = 'https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_L_in21k.pyth' ,
995
+ hf_hub_id = 'timm/' ,
989
996
num_classes = 19168 ),
990
997
'mvitv2_huge_cls.fb_inw21k' : _cfg (
991
998
url = 'https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_H_in21k.pyth' ,
999
+ hf_hub_id = 'timm/' ,
992
1000
num_classes = 19168 ),
993
1001
})
994
1002
0 commit comments