Spaces:
Paused
Paused
Update app_gradio.py
Browse files- 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)]
|
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)
|