rynmurdock commited on
Commit
1d16802
·
verified ·
1 Parent(s): a5ec0b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ def predict(
51
  im_emb = torch.zeros(1, 1280, dtype=torch.float16, device='cuda')
52
  image = pipe(
53
  prompt=prompt,
54
- ip_adapter_emb=im_emb,
55
  height=1024,
56
  width=1024,
57
  num_inference_steps=8,
 
51
  im_emb = torch.zeros(1, 1280, dtype=torch.float16, device='cuda')
52
  image = pipe(
53
  prompt=prompt,
54
+ ip_adapter_emb=im_emb.to('cuda'),
55
  height=1024,
56
  width=1024,
57
  num_inference_steps=8,