Skip to content

Commit 992eb6c

Browse files
committed
Add MetaCLIP ViT-H-14 AllTogether (v1.2) weight
1 parent aeaf2a0 commit 992eb6c

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/open_clip/pretrained.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -318,6 +318,11 @@ def _mccfg(url='', hf_hub='', **kwargs):
318318
hf_hub="timm/vit_huge_patch14_clip_224.metaclip_2pt5b/",
319319
quick_gelu=True,
320320
),
321+
metaclip_alltogether=_pcfg(
322+
url="https://dl.fbaipublicfiles.com/MMPT/metaclip/h14_v1.2_altogether.pt",
323+
hf_hub="timm/vit_huge_patch14_clip_224.metaclip_alltogether/",
324+
# NOTE unlike other MetaCLIP models, this is not using QuickGELU, yay!
325+
),
321326
# DFN-5B (quick-gelu)
322327
dfn5b=_pcfg(
323328
hf_hub='apple/DFN5B-CLIP-ViT-H-14/',

0 commit comments

Comments
 (0)