Linoy Tsaban
commited on
Commit
·
7ff8616
1
Parent(s):
9156300
Update app.py
Browse files
app.py
CHANGED
|
@@ -129,12 +129,12 @@ def invert_and_reconstruct(
|
|
| 129 |
|
| 130 |
if do_inversion:
|
| 131 |
# invert and retrieve noise maps and latent
|
| 132 |
-
zs_tensor, wts_tensor = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps, cfg_scale_src=
|
| 133 |
wts = gr.State(value=wts_tensor)
|
| 134 |
zs = gr.State(value=zs_tensor)
|
| 135 |
do_inversion = False
|
| 136 |
|
| 137 |
-
output = sample(zs.value, wts.value, prompt_tar=tar_prompt, skip=skip, cfg_scale_tar=
|
| 138 |
|
| 139 |
return output, wts, zs, do_inversion
|
| 140 |
|
|
|
|
| 129 |
|
| 130 |
if do_inversion:
|
| 131 |
# invert and retrieve noise maps and latent
|
| 132 |
+
zs_tensor, wts_tensor = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps, cfg_scale_src=src_cfg_scale)
|
| 133 |
wts = gr.State(value=wts_tensor)
|
| 134 |
zs = gr.State(value=zs_tensor)
|
| 135 |
do_inversion = False
|
| 136 |
|
| 137 |
+
output = sample(zs.value, wts.value, prompt_tar=tar_prompt, skip=skip, cfg_scale_tar=tar_cfg_scale)
|
| 138 |
|
| 139 |
return output, wts, zs, do_inversion
|
| 140 |
|