swap to a better clip model

ShoukanLabs · Oct 9, 2023 · 883c3ed · 883c3ed
1 parent 1574763
commit 883c3ed
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train_muse_maskgit.py b/train_muse_maskgit.py
@@ -1003,8 +1003,8 @@ def main():
             clip_precision = args.mixed_precision
 
         clip = open_clip.create_model_and_transforms(
-            "ViT-B-32-quickgelu",
-            pretrained="metaclip/b32_400m.pt",
+            "ViT-L-14",
+            pretrained="metaclip/l14_400m.pt",
             cache_dir=args.cache_path,
             precision=clip_precision,
             device=accelerator.device,