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 |
|