Linoy Tsaban
commited on
Commit
·
9df32ef
1
Parent(s):
87c0592
Update app.py
Browse files
app.py
CHANGED
|
@@ -124,7 +124,7 @@ def invert_and_reconstruct(
|
|
| 124 |
):
|
| 125 |
offsets=(0,0,0,0)
|
| 126 |
torch.manual_seed(seed)
|
| 127 |
-
x0 = load_512(input_image,
|
| 128 |
|
| 129 |
|
| 130 |
# invert
|
|
@@ -163,7 +163,7 @@ def edit(input_image,
|
|
| 163 |
|
| 164 |
|
| 165 |
offsets = (0,0,0,0)
|
| 166 |
-
x0 = load_512(input_image,
|
| 167 |
|
| 168 |
# invert
|
| 169 |
# wt, zs, wts = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps, cfg_scale_src=src_cfg_scale)
|
|
|
|
| 124 |
):
|
| 125 |
offsets=(0,0,0,0)
|
| 126 |
torch.manual_seed(seed)
|
| 127 |
+
x0 = load_512(input_image, device=device)
|
| 128 |
|
| 129 |
|
| 130 |
# invert
|
|
|
|
| 163 |
|
| 164 |
|
| 165 |
offsets = (0,0,0,0)
|
| 166 |
+
x0 = load_512(input_image, device=device)
|
| 167 |
|
| 168 |
# invert
|
| 169 |
# wt, zs, wts = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps, cfg_scale_src=src_cfg_scale)
|