Skip to content

Commit d9321b0

Browse files
committed
Add weights for fine-tuned siglip so400m. Add webli_i18n pretrained tags for the multi-lingual model variants (incl older base)
1 parent 01b6226 commit d9321b0

File tree

1 file changed

+14
-6
lines changed

1 file changed

+14
-6
lines changed

timm/models/vision_transformer.py

Lines changed: 14 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1817,6 +1817,11 @@ def _cfg(url: str = '', **kwargs) -> Dict[str, Any]:
18171817
hf_hub_filename='open_clip_pytorch_model.bin',
18181818
input_size=(3, 256, 256),
18191819
num_classes=0),
1820+
'vit_base_patch16_siglip_256.webli_i18n': _cfg(
1821+
hf_hub_id='timm/ViT-B-16-SigLIP-i18n-256',
1822+
hf_hub_filename='open_clip_pytorch_model.bin',
1823+
input_size=(3, 256, 256),
1824+
num_classes=0),
18201825
'vit_base_patch16_siglip_384.webli': _cfg(
18211826
hf_hub_id='timm/ViT-B-16-SigLIP-384',
18221827
hf_hub_filename='open_clip_pytorch_model.bin',
@@ -1841,7 +1846,7 @@ def _cfg(url: str = '', **kwargs) -> Dict[str, Any]:
18411846
hf_hub_id='timm/ViT-SO400M-14-SigLIP',
18421847
hf_hub_filename='open_clip_pytorch_model.bin',
18431848
num_classes=0),
1844-
'vit_so400m_patch16_siglip_256.webli': _cfg(
1849+
'vit_so400m_patch16_siglip_256.webli_i18n': _cfg(
18451850
hf_hub_id='timm/ViT-SO400M-16-SigLIP-i18n-256',
18461851
hf_hub_filename='open_clip_pytorch_model.bin',
18471852
input_size=(3, 256, 256),
@@ -1866,6 +1871,11 @@ def _cfg(url: str = '', **kwargs) -> Dict[str, Any]:
18661871
hf_hub_filename='open_clip_pytorch_model.bin',
18671872
input_size=(3, 256, 256),
18681873
num_classes=0),
1874+
'vit_base_patch16_siglip_gap_256.webli_i18n': _cfg(
1875+
hf_hub_id='timm/ViT-B-16-SigLIP-i18n-256',
1876+
hf_hub_filename='open_clip_pytorch_model.bin',
1877+
input_size=(3, 256, 256),
1878+
num_classes=0),
18691879
'vit_base_patch16_siglip_gap_384.webli': _cfg(
18701880
hf_hub_id='timm/ViT-B-16-SigLIP-384',
18711881
hf_hub_filename='open_clip_pytorch_model.bin',
@@ -1900,7 +1910,7 @@ def _cfg(url: str = '', **kwargs) -> Dict[str, Any]:
19001910
hf_hub_filename='paligemma-3b-pt-224.npz',
19011911
custom_load='hf',
19021912
num_classes=0),
1903-
'vit_so400m_patch16_siglip_gap_256.webli': _cfg(
1913+
'vit_so400m_patch16_siglip_gap_256.webli_i18n': _cfg(
19041914
hf_hub_id='timm/ViT-SO400M-16-SigLIP-i18n-256',
19051915
hf_hub_filename='open_clip_pytorch_model.bin',
19061916
input_size=(3, 256, 256),
@@ -1935,13 +1945,11 @@ def _cfg(url: str = '', **kwargs) -> Dict[str, Any]:
19351945
num_classes=0),
19361946

19371947
'vit_so400m_patch14_siglip_378.webli_ft_in1k': _cfg(
1938-
#hf_hub_id='timm/',
1939-
#file='vit_so400m_p14_378_map-8.pth',
1948+
hf_hub_id='timm/',
19401949
input_size=(3, 378, 378), crop_pct=1.0, crop_mode='squash',
19411950
),
19421951
'vit_so400m_patch14_siglip_gap_378.webli_ft_in1k': _cfg(
1943-
# hf_hub_id='timm/',
1944-
#file='vit_so400m_p14_378_gap-8.pth',
1952+
hf_hub_id='timm/',
19451953
input_size=(3, 378, 378), crop_pct=1.0, crop_mode='squash',
19461954
),
19471955

0 commit comments

Comments
 (0)