Spaces:
Sleeping
Sleeping
rynmurdock
commited on
Update app.py
Browse files
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,
|