Spaces:
Running
on
A10G
Running
on
A10G
rynmurdock
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -49,8 +49,6 @@ def predict(
|
|
49 |
with torch.no_grad():
|
50 |
if im_emb == None:
|
51 |
im_emb = torch.zeros(1, 1280, dtype=torch.float16, device='cuda')
|
52 |
-
else:
|
53 |
-
im_emb = torch.tensor([float(i) for i in im_emb.split(', ')]).unsqueeze(0).to(dtype=torch.float16).to('cuda')
|
54 |
image = pipe(
|
55 |
prompt=prompt,
|
56 |
ip_adapter_emb=im_emb,
|
|
|
49 |
with torch.no_grad():
|
50 |
if im_emb == None:
|
51 |
im_emb = torch.zeros(1, 1280, dtype=torch.float16, device='cuda')
|
|
|
|
|
52 |
image = pipe(
|
53 |
prompt=prompt,
|
54 |
ip_adapter_emb=im_emb,
|