Spaces:
Running
on
A10G
Running
on
A10G
Linoy Tsaban
commited on
Commit
•
09eb804
1
Parent(s):
7ff8616
Update app.py
Browse files
app.py
CHANGED
@@ -28,7 +28,7 @@ def invert(x0, prompt_src="", num_diffusion_steps=100, cfg_scale_src = 3.5, eta
|
|
28 |
|
29 |
# find Zs and wts - forward process
|
30 |
wt, zs, wts = inversion_forward_process(sd_pipe, w0, etas=eta, prompt=prompt_src, cfg_scale=cfg_scale_src, prog_bar=True, num_inference_steps=num_diffusion_steps)
|
31 |
-
return
|
32 |
|
33 |
|
34 |
|
|
|
28 |
|
29 |
# find Zs and wts - forward process
|
30 |
wt, zs, wts = inversion_forward_process(sd_pipe, w0, etas=eta, prompt=prompt_src, cfg_scale=cfg_scale_src, prog_bar=True, num_inference_steps=num_diffusion_steps)
|
31 |
+
return zs, wts
|
32 |
|
33 |
|
34 |
|