From 5d599a6a106d4872d6ef12364c9c9f5637f820ec Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 25 Aug 2023 10:39:02 -0700 Subject: [PATCH] RepViT weights on HF hub --- timm/models/repvit.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/timm/models/repvit.py b/timm/models/repvit.py index 55b6ba4bea..a0def2f41c 100644 --- 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' ), } )