Skip to content

Commit 3cc4d7a

Browse files
committed
Fix missing register for 224 eva model
1 parent eba07b0 commit 3cc4d7a

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

timm/models/beit.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -130,18 +130,15 @@ def _cfg(url='', **kwargs):
130130
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD,
131131
),
132132
'eva_giant_patch14_336.clip_ft_in1k': _cfg(
133-
hf_hub_id='BAAI/EVA',
134-
hf_hub_filename='eva_clip_vis_enc_sz336_ftcls_89p4.pt',
133+
hf_hub_id='BAAI/EVA', hf_hub_filename='eva_clip_vis_enc_sz336_ftcls_89p4.pt',
135134
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD,
136135
input_size=(3, 336, 336)),
137136
'eva_giant_patch14_336.m30m_ft_in22k_in1k': _cfg(
138-
hf_hub_id='BAAI/EVA',
139-
hf_hub_filename='eva_21k_1k_336px_psz14_ema_89p6.pt',
137+
hf_hub_id='BAAI/EVA', hf_hub_filename='eva_21k_1k_336px_psz14_ema_89p6.pt',
140138
mean=IMAGENET_DEFAULT_MEAN, std=IMAGENET_DEFAULT_STD,
141139
input_size=(3, 336, 336)),
142140
'eva_giant_patch14_560.m30m_ft_in22k_in1k': _cfg(
143-
hf_hub_id='BAAI/EVA',
144-
hf_hub_filename='eva_21k_1k_560px_psz14_ema_89p7.pt',
141+
hf_hub_id='BAAI/EVA', hf_hub_filename='eva_21k_1k_560px_psz14_ema_89p7.pt',
145142
mean=IMAGENET_DEFAULT_MEAN, std=IMAGENET_DEFAULT_STD,
146143
input_size=(3, 560, 560)),
147144
})
@@ -537,6 +534,7 @@ def beitv2_large_patch16_224_in22k(pretrained=False, **kwargs):
537534
return model
538535

539536

537+
@register_model
540538
def eva_giant_patch14_224(pretrained=False, **kwargs):
541539
""" EVA-g model https://arxiv.org/abs/2211.07636 """
542540
model_kwargs = dict(

0 commit comments

Comments
 (0)