Skip to content

Commit 77bf914

Browse files
committed
Add weights for siglip2 naflex vit encoders
1 parent dddb914 commit 77bf914

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

timm/models/naflexvit.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1921,8 +1921,13 @@ def _cfg(url: str = '', **kwargs) -> Dict[str, Any]:
19211921
),
19221922
'naflexvit_base_patch16_map.untrained': _cfg(),
19231923

1924-
'naflexvit_base_patch16_siglip.untrained': _cfg(),
1925-
'naflexvit_so400m_patch16_siglip.untrained': _cfg(),
1924+
# SigLIP-2 NaFlex vit encoder weights
1925+
'naflexvit_base_patch16_siglip.v2_webli': _cfg(
1926+
hf_hub_id='timm/',
1927+
num_classes=0),
1928+
'naflexvit_so400m_patch16_siglip.v2_webli': _cfg(
1929+
hf_hub_id='timm/',
1930+
num_classes=0),
19261931
})
19271932

19281933

0 commit comments

Comments
 (0)