@@ -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
540538def eva_giant_patch14_224 (pretrained = False , ** kwargs ):
541539 """ EVA-g model https://arxiv.org/abs/2211.07636 """
542540 model_kwargs = dict (
0 commit comments