Spaces:

rynmurdock
/

generative_recsys

Running on A10G

rynmurdock commited on Apr 11

Commit

32c5eea

•

1 Parent(s): 2c5d6c7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -57,7 +57,7 @@ def predict(
             im_emb = torch.zeros(1, 1280, dtype=torch.float16, device='cuda')
         image = pipe(
             prompt=prompt,
-            ip_adapter_emb=im_emb.to('cuda'),
             height=1024,
             width=1024,
             num_inference_steps=2,

             im_emb = torch.zeros(1, 1280, dtype=torch.float16, device='cuda')
         image = pipe(
             prompt=prompt,
+            ip_adapter_emb=[im_emb.to('cuda')],
             height=1024,
             width=1024,
             num_inference_steps=2,