Skip to content

Commit f7aa0a1

Browse files
committed
Add missing vit_wee weight
1 parent 49de391 commit f7aa0a1

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

timm/models/vision_transformer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1791,6 +1791,7 @@ def _cfg(url: str = '', **kwargs) -> Dict[str, Any]:
17911791
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, num_classes=512),
17921792

17931793
'vit_wee_patch16_reg1_gap_256.sbb_in1k': _cfg(
1794+
hf_hub_id='timm/',
17941795
input_size=(3, 256, 256), crop_pct=0.95),
17951796
'vit_pwee_patch16_reg1_gap_256.sbb_in1k': _cfg(
17961797
hf_hub_id='timm/',

0 commit comments

Comments
 (0)