diff --git a/timm/models/naflexvit.py b/timm/models/naflexvit.py index 94fe8fa63d..f8e27b5044 100644 --- a/timm/models/naflexvit.py +++ b/timm/models/naflexvit.py @@ -1921,8 +1921,13 @@ def _cfg(url: str = '', **kwargs) -> Dict[str, Any]: ), 'naflexvit_base_patch16_map.untrained': _cfg(), - 'naflexvit_base_patch16_siglip.untrained': _cfg(), - 'naflexvit_so400m_patch16_siglip.untrained': _cfg(), + # SigLIP-2 NaFlex vit encoder weights + 'naflexvit_base_patch16_siglip.v2_webli': _cfg( + hf_hub_id='timm/', + num_classes=0), + 'naflexvit_so400m_patch16_siglip.v2_webli': _cfg( + hf_hub_id='timm/', + num_classes=0), })