From 76df2ff0d7d5ade55f5954bfc5827c3a36dd90aa Mon Sep 17 00:00:00 2001 From: Gabriel Ilharco Date: Fri, 29 Sep 2023 22:36:48 +0000 Subject: [PATCH] Add B/32 and B/16 trained on DC-1B --- src/open_clip/pretrained.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/open_clip/pretrained.py b/src/open_clip/pretrained.py index 1465a2325..728755a2e 100644 --- a/src/open_clip/pretrained.py +++ b/src/open_clip/pretrained.py @@ -84,6 +84,8 @@ def _pcfg(url='', hf_hub='', mean=None, std=None): laion2b_e16=_pcfg( "https://github.com/mlfoundations/open_clip/releases/download/v0.2-weights/vit_b_32-laion2b_e16-af8dbd0c.pth"), laion2b_s34b_b79k=_pcfg(hf_hub='laion/CLIP-ViT-B-32-laion2B-s34B-b79K/'), + # DataComp-XL models + datacomp_xl_s13b_b90k=_pcfg(hf_hub='laion/CLIP-ViT-B-32-DataComp.XL-s13B-b90K/'), # DataComp-M models datacomp_m_s128m_b4k=_pcfg(hf_hub='laion/CLIP-ViT-B-32-DataComp.M-s128M-b4K/'), commonpool_m_clip_s128m_b4k=_pcfg(hf_hub='laion/CLIP-ViT-B-32-CommonPool.M.clip-s128M-b4K/'), @@ -119,6 +121,8 @@ def _pcfg(url='', hf_hub='', mean=None, std=None): laion400m_e32=_pcfg( "https://github.com/mlfoundations/open_clip/releases/download/v0.2-weights/vit_b_16-laion400m_e32-55e67d44.pt"), laion2b_s34b_b88k=_pcfg(hf_hub='laion/CLIP-ViT-B-16-laion2B-s34B-b88K/'), + # DataComp-XL models + datacomp_xl_s13b_b90k=_pcfg(hf_hub='laion/CLIP-ViT-B-16-DataComp.XL-s13B-b90K/'), # DataComp-L models datacomp_l_s1b_b8k=_pcfg(hf_hub='laion/CLIP-ViT-B-16-DataComp.L-s1B-b8K/'), commonpool_l_clip_s1b_b8k=_pcfg(hf_hub='laion/CLIP-ViT-B-16-CommonPool.L.clip-s1B-b8K/'),