From f7aa0a1a71bc728ac7734655a8e0476f43a4fe73 Mon Sep 17 00:00:00 2001
From: Ross Wightman <rwightman@gmail.com>
Date: Mon, 13 May 2024 12:05:47 -0700
Subject: [PATCH] Add missing vit_wee weight

---
 timm/models/vision_transformer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/timm/models/vision_transformer.py b/timm/models/vision_transformer.py
index 5fbabb59..df9aa35c 100644
--- a/timm/models/vision_transformer.py
+++ b/timm/models/vision_transformer.py
@@ -1791,6 +1791,7 @@ default_cfgs = {
         mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD, num_classes=512),
 
     'vit_wee_patch16_reg1_gap_256.sbb_in1k': _cfg(
+        hf_hub_id='timm/',
         input_size=(3, 256, 256), crop_pct=0.95),
     'vit_pwee_patch16_reg1_gap_256.sbb_in1k': _cfg(
         hf_hub_id='timm/',