From 0719dc3df33622dfe0047fda3b66a7f4b0f1b490 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 4 Aug 2025 11:57:59 -0700 Subject: [PATCH] Fix #2562, add siglip2 naflex vit encoder weights --- timm/models/naflexvit.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) 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), })