Upload models/backbone.py with huggingface_hub

models/backbone.py

@@ -185,7 +185,7 @@ def build_backbone(args):
                 'swin_L_384_22k': 'swin_large_patch4_window12_384_22k.pth',
             }
             pretrainedpath = os.path.join(pretrained_dir, PTDICT[args.backbone])
-            checkpoint = torch.load(pretrainedpath, map_location='cpu')['model']
+            checkpoint = torch.load(pretrainedpath, map_location='cpu', weights_only=False)['model']
             from collections import OrderedDict
             def key_select_function(keyname):
                 if 'head' in keyname: