RepViT weights on HF hub

huggingface · Aug 25, 2023 · 5d599a6 · 5d599a6
1 parent 56c2854
commit 5d599a6
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/timm/models/repvit.py b/timm/models/repvit.py
@@ -368,13 +368,16 @@ def _cfg(url='', **kwargs):
 default_cfgs = generate_default_cfgs(
     {
         'repvit_m1.dist_in1k': _cfg(
-            url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m1_distill_300_timm.pth'
+            hf_hub_id='timm/',
+            # url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m1_distill_300_timm.pth'
         ),
         'repvit_m2.dist_in1k': _cfg(
-            url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m2_distill_300_timm.pth'
+            hf_hub_id='timm/',
+            # url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m2_distill_300_timm.pth'
         ),
         'repvit_m3.dist_in1k': _cfg(
-            url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m3_distill_300_timm.pth'
+            hf_hub_id='timm/',
+            # url='https://github.com/THU-MIG/RepViT/releases/download/v1.0/repvit_m3_distill_300_timm.pth'
         ),
     }
 )