fffiloni commited on
Commit
618f8ef
1 Parent(s): be63a84

Update app_gradio.py

Browse files
Files changed (1) hide show
  1. app_gradio.py +1 -1
app_gradio.py CHANGED
@@ -67,7 +67,7 @@ def prepare_latents(pipe, x_aug):
67
 
68
  @torch.no_grad()
69
  def invert(pipe, inv, load_name, device="cuda", dtype=torch.bfloat16):
70
- input_img = [load_image(load_name, 256).to(device, dtype=dtype).unsqueeze(1)] * 8
71
  input_img = torch.cat(input_img, dim=1)
72
  latents = prepare_latents(pipe, input_img).to(torch.bfloat16)
73
  inv.set_timesteps(25)
 
67
 
68
  @torch.no_grad()
69
  def invert(pipe, inv, load_name, device="cuda", dtype=torch.bfloat16):
70
+ input_img = [load_image(load_name, 256).to(device, dtype=dtype).unsqueeze(1)]
71
  input_img = torch.cat(input_img, dim=1)
72
  latents = prepare_latents(pipe, input_img).to(torch.bfloat16)
73
  inv.set_timesteps(25)