Keshabwi66 commited on
Commit
c4c66c7
·
1 Parent(s): 2d8cc52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=prompt_embedsto.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),
 
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),