rynmurdock commited on
Commit
37458c6
1 Parent(s): 32c5eea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ def predict(
57
  im_emb = torch.zeros(1, 1280, dtype=torch.float16, device='cuda')
58
  image = pipe(
59
  prompt=prompt,
60
- ip_adapter_emb=[im_emb.to('cuda')],
61
  height=1024,
62
  width=1024,
63
  num_inference_steps=2,
 
57
  im_emb = torch.zeros(1, 1280, dtype=torch.float16, device='cuda')
58
  image = pipe(
59
  prompt=prompt,
60
+ image_embeds=[im_emb.to('cuda')],
61
  height=1024,
62
  width=1024,
63
  num_inference_steps=2,