add ViTamin CVPR 2024

mlfoundations · May 5, 2024 · 48e25d0 · 48e25d0
1 parent 9eaf242
commit 48e25d0
Show file tree

Hide file tree

Showing 2 changed files with 562 additions and 1 deletion.
diff --git a/src/open_clip/timm_model.py b/src/open_clip/timm_model.py
@@ -23,7 +23,7 @@
     timm = None
 
 from .utils import freeze_batch_norm_2d
-
+from .vitamin import *
 
 class TimmModel(nn.Module):
     """ timm model adapter