Spaces:
Runtime error
Runtime error
Commit
·
c4c66c7
1
Parent(s):
2d8cc52
Update app.py
Browse files
app.py
CHANGED
|
@@ -198,7 +198,7 @@ def generate_virtual_try_on(person_image, cloth_image, mask_image, pose_image,cl
|
|
| 198 |
|
| 199 |
with torch.no_grad():
|
| 200 |
images = pipe(
|
| 201 |
-
prompt_embeds=
|
| 202 |
negative_prompt_embeds=negative_prompt_embeds.to(device,torch.float16),
|
| 203 |
pooled_prompt_embeds=pooled_prompt_embeds.to(device,torch.float16),
|
| 204 |
negative_pooled_prompt_embeds=negative_pooled_prompt_embeds.to(device,torch.float16),
|
|
|
|
| 198 |
|
| 199 |
with torch.no_grad():
|
| 200 |
images = pipe(
|
| 201 |
+
prompt_embeds=prompt_embeds.to(device,torch.float16),
|
| 202 |
negative_prompt_embeds=negative_prompt_embeds.to(device,torch.float16),
|
| 203 |
pooled_prompt_embeds=pooled_prompt_embeds.to(device,torch.float16),
|
| 204 |
negative_pooled_prompt_embeds=negative_pooled_prompt_embeds.to(device,torch.float16),
|