From 3752bb6755386c08b09f9700b9749bca1158622d Mon Sep 17 00:00:00 2001 From: Desai Xie <32966374+desaixie@users.noreply.github.com> Date: Fri, 27 Sep 2024 22:47:02 -0700 Subject: [PATCH] New VideoMAEv2 ckpt_url --- cdfvd/third_party/VideoMAEv2/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cdfvd/third_party/VideoMAEv2/utils.py b/cdfvd/third_party/VideoMAEv2/utils.py index 0506435..7ac7ec4 100644 --- a/cdfvd/third_party/VideoMAEv2/utils.py +++ b/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) \ No newline at end of file + return model.to(device)