zideliu commited on
Commit
28de44c
1 Parent(s): 815b624

Update clip

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -87,7 +87,7 @@ device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
87
  print(device)
88
  # Load open_clip and vq model
89
  print("GPU memory:",torch.cuda.memory_allocated(0)/1024/1024/1024,"GB")
90
- prompt_model,_,_ = open_clip.create_model_and_transforms('ViT-bigG-14', 'laion2b_s39b_b160k',device=device)
91
  print("GPU memory:",torch.cuda.memory_allocated(0)/1024/1024/1024,"GB")
92
 
93
  prompt_model = prompt_model.to(device)
 
87
  print(device)
88
  # Load open_clip and vq model
89
  print("GPU memory:",torch.cuda.memory_allocated(0)/1024/1024/1024,"GB")
90
+ prompt_model,_,_ = open_clip.create_model_and_transforms('ViT-bigG-14', 'laion2b_s39b_b160k',precision='fp16',device=device)
91
  print("GPU memory:",torch.cuda.memory_allocated(0)/1024/1024/1024,"GB")
92
 
93
  prompt_model = prompt_model.to(device)