New VideoMAEv2 ckpt_url

cdfvd/third_party/VideoMAEv2/utils.py

@@ -52,7 +52,7 @@ def load_videomae_model(device, ckpt_path=None):
 
     if not os.path.exists(ckpt_path):
         # download the ckpt to the path
-        ckpt_url = 'https://pjlab-gvm-data.oss-cn-shanghai.aliyuncs.com/internvideo/videomaev2/vit_g_hybrid_pt_1200e_ssv2_ft.pth'
+        ckpt_url = 'https://huggingface.co/OpenGVLab/InternVideoMAE_models/resolve/main/mae-g/vit_g_hybrid_pt_1200e_ssv2_ft.pth'
         response = requests.get(ckpt_url, stream=True, allow_redirects=True)
         total_size = int(response.headers.get("content-length", 0))
         block_size = 1024
@@ -78,4 +78,4 @@ def load_videomae_model(device, ckpt_path=None):
             ckpt = ckpt[model_key]
             break
     model.load_state_dict(ckpt)
-    return model.to(device)
+    return model.to(device)