Update clip

app.py

@@ -25,14 +25,14 @@ empty_context = np.load("assets/contexts/empty_context.npy")
 
 print("downloading cc3m-285000.ckpt")
 os.makedirs("assets/ckpts/cc3m-285000.ckpt",exist_ok=True)
-os.system("ls")
+
 wget.download("https://huggingface.co/nzl-thu/MUSE/resolve/main/assets/ckpts/cc3m-285000.ckpt/lr_scheduler.pth","assets/ckpts/cc3m-285000.ckpt/lr_scheduler.pth")
 wget.download("https://huggingface.co/nzl-thu/MUSE/resolve/main/assets/ckpts/cc3m-285000.ckpt/optimizer.pth","assets/ckpts/cc3m-285000.ckpt/optimizer.pth")
 wget.download("https://huggingface.co/nzl-thu/MUSE/resolve/main/assets/ckpts/cc3m-285000.ckpt/nnet.pth","assets/ckpts/cc3m-285000.ckpt/nnet.pth")
 wget.download("https://huggingface.co/nzl-thu/MUSE/resolve/main/assets/ckpts/cc3m-285000.ckpt/nnet_ema.pth","assets/ckpts/cc3m-285000.ckpt/nnet_ema.pth")
 wget.download("https://huggingface.co/nzl-thu/MUSE/resolve/main/assets/ckpts/cc3m-285000.ckpt/step.pth","assets/ckpts/cc3m-285000.ckpt/step.pth")
 wget.download("https://huggingface.co/zideliu/vqgan/resolve/main/vqgan_jax_strongaug.ckpt","assets/vqgan_jax_strongaug.ckpt")
-
+os.system("ls assets/ckpts/cc3m-285000.ckpt")
 def set_seed(seed: int):
     random.seed(seed)
     np.random.seed(seed)
@@ -115,9 +115,9 @@ def unprocess(x):
 
 config = get_config()
 device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
-
+print(device)
 # Load open_clip and vq model
-prompt_model,_,_ = open_clip.create_model_and_transforms('ViT-bigG-14', 'laion2b_s39b_b160k')
+prompt_model,_,_ = open_clip.create_model_and_transforms('ViT-bigG-14', 'laion2b_s39b_b160k',device='cuda')
 prompt_model = prompt_model.to(device)
 prompt_model.eval()
 tokenizer = open_clip.get_tokenizer('ViT-bigG-14')