johnberg commited on
Commit
f91b53c
·
1 Parent(s): 19ed308

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ adapter = CLIPAdapterWithDecoder(opts)
50
  adapter.eval()
51
  adapter.to(device)
52
 
53
- clip_model, _ = clip.load("ViT-B/32", device='cuda')
54
 
55
  def manipulate(input_image, caption):
56
  aligned_image = run_alignment(input_image)
 
50
  adapter.eval()
51
  adapter.to(device)
52
 
53
+ clip_model, _ = clip.load("ViT-B/32", device=device)
54
 
55
  def manipulate(input_image, caption):
56
  aligned_image = run_alignment(input_image)