mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Add weights for fine-tuned siglip so400m. Add webli_i18n pretrained tags for the multi-lingual model variants (incl older base)

Browse Source
This commit is contained in:
Ross Wightman 2024-10-09 09:04:44 -07:00
parent 01b62264af
commit d9321b0e10
1 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
timm/models/vision_transformer.py
View File

@ -1817,6 +1817,11 @@ default_cfgs = {
hf_hub_filename='open_clip_pytorch_model.bin', hf_hub_filename='open_clip_pytorch_model.bin',
input_size=(3, 256, 256), input_size=(3, 256, 256),
num_classes=0), num_classes=0),
'vit_base_patch16_siglip_256.webli_i18n': _cfg(
hf_hub_id='timm/ViT-B-16-SigLIP-i18n-256',
hf_hub_filename='open_clip_pytorch_model.bin',
input_size=(3, 256, 256),
num_classes=0),
'vit_base_patch16_siglip_384.webli': _cfg( 'vit_base_patch16_siglip_384.webli': _cfg(
hf_hub_id='timm/ViT-B-16-SigLIP-384', hf_hub_id='timm/ViT-B-16-SigLIP-384',
hf_hub_filename='open_clip_pytorch_model.bin', hf_hub_filename='open_clip_pytorch_model.bin',
@ -1841,7 +1846,7 @@ default_cfgs = {
hf_hub_id='timm/ViT-SO400M-14-SigLIP', hf_hub_id='timm/ViT-SO400M-14-SigLIP',
hf_hub_filename='open_clip_pytorch_model.bin', hf_hub_filename='open_clip_pytorch_model.bin',
num_classes=0), num_classes=0),
'vit_so400m_patch16_siglip_256.webli': _cfg( 'vit_so400m_patch16_siglip_256.webli_i18n': _cfg(
hf_hub_id='timm/ViT-SO400M-16-SigLIP-i18n-256', hf_hub_id='timm/ViT-SO400M-16-SigLIP-i18n-256',
hf_hub_filename='open_clip_pytorch_model.bin', hf_hub_filename='open_clip_pytorch_model.bin',
input_size=(3, 256, 256), input_size=(3, 256, 256),
@ -1866,6 +1871,11 @@ default_cfgs = {
hf_hub_filename='open_clip_pytorch_model.bin', hf_hub_filename='open_clip_pytorch_model.bin',
input_size=(3, 256, 256), input_size=(3, 256, 256),
num_classes=0), num_classes=0),
'vit_base_patch16_siglip_gap_256.webli_i18n': _cfg(
hf_hub_id='timm/ViT-B-16-SigLIP-i18n-256',
hf_hub_filename='open_clip_pytorch_model.bin',
input_size=(3, 256, 256),
num_classes=0),
'vit_base_patch16_siglip_gap_384.webli': _cfg( 'vit_base_patch16_siglip_gap_384.webli': _cfg(
hf_hub_id='timm/ViT-B-16-SigLIP-384', hf_hub_id='timm/ViT-B-16-SigLIP-384',
hf_hub_filename='open_clip_pytorch_model.bin', hf_hub_filename='open_clip_pytorch_model.bin',
@ -1900,7 +1910,7 @@ default_cfgs = {
hf_hub_filename='paligemma-3b-pt-224.npz', hf_hub_filename='paligemma-3b-pt-224.npz',
custom_load='hf', custom_load='hf',
num_classes=0), num_classes=0),
'vit_so400m_patch16_siglip_gap_256.webli': _cfg( 'vit_so400m_patch16_siglip_gap_256.webli_i18n': _cfg(
hf_hub_id='timm/ViT-SO400M-16-SigLIP-i18n-256', hf_hub_id='timm/ViT-SO400M-16-SigLIP-i18n-256',
hf_hub_filename='open_clip_pytorch_model.bin', hf_hub_filename='open_clip_pytorch_model.bin',
input_size=(3, 256, 256), input_size=(3, 256, 256),
@ -1935,13 +1945,11 @@ default_cfgs = {
num_classes=0), num_classes=0),
'vit_so400m_patch14_siglip_378.webli_ft_in1k': _cfg( 'vit_so400m_patch14_siglip_378.webli_ft_in1k': _cfg(
#hf_hub_id='timm/', hf_hub_id='timm/',
#file='vit_so400m_p14_378_map-8.pth',
input_size=(3, 378, 378), crop_pct=1.0, crop_mode='squash', input_size=(3, 378, 378), crop_pct=1.0, crop_mode='squash',
), ),
'vit_so400m_patch14_siglip_gap_378.webli_ft_in1k': _cfg( 'vit_so400m_patch14_siglip_gap_378.webli_ft_in1k': _cfg(
# hf_hub_id='timm/', hf_hub_id='timm/',
#file='vit_so400m_p14_378_gap-8.pth',
input_size=(3, 378, 378), crop_pct=1.0, crop_mode='squash', input_size=(3, 378, 378), crop_pct=1.0, crop_mode='squash',
), ),