@@ -975,20 +975,28 @@ def _cfg(url='', **kwargs):
975975
976976
977977default_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/' ),
982987
983988 'mvitv2_small_cls' : _cfg (url = '' ),
984989 'mvitv2_base_cls.fb_inw21k' : _cfg (
985990 url = 'https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_B_in21k.pyth' ,
991+ hf_hub_id = 'timm/' ,
986992 num_classes = 19168 ),
987993 'mvitv2_large_cls.fb_inw21k' : _cfg (
988994 url = 'https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_L_in21k.pyth' ,
995+ hf_hub_id = 'timm/' ,
989996 num_classes = 19168 ),
990997 'mvitv2_huge_cls.fb_inw21k' : _cfg (
991998 url = 'https://dl.fbaipublicfiles.com/mvit/mvitv2_models/MViTv2_H_in21k.pyth' ,
999+ hf_hub_id = 'timm/' ,
9921000 num_classes = 19168 ),
9931001})
9941002
0 commit comments