Linoy Tsaban
commited on
Commit
•
774cc5f
1
Parent(s):
71e0668
Update app.py
Browse files
app.py
CHANGED
@@ -85,11 +85,10 @@ def edit(input_image,
|
|
85 |
wt, zs, wts = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps)
|
86 |
latnets = wts[skip].expand(1, -1, -1, -1)
|
87 |
|
88 |
-
|
89 |
#pure DDPM output
|
90 |
pure_ddpm_out = sample(wt, zs, wts, prompt_tar=tar_prompt,
|
91 |
-
cfg_scale_tar=tar_cfg_scale, skip=skip
|
92 |
-
eta = eta)
|
93 |
yield pure_ddpm_out, pure_ddpm_out
|
94 |
|
95 |
editing_args = dict(
|
|
|
85 |
wt, zs, wts = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps)
|
86 |
latnets = wts[skip].expand(1, -1, -1, -1)
|
87 |
|
88 |
+
|
89 |
#pure DDPM output
|
90 |
pure_ddpm_out = sample(wt, zs, wts, prompt_tar=tar_prompt,
|
91 |
+
cfg_scale_tar=tar_cfg_scale, skip=skip)
|
|
|
92 |
yield pure_ddpm_out, pure_ddpm_out
|
93 |
|
94 |
editing_args = dict(
|