Linoy Tsaban
commited on
Commit
·
6a2fd41
1
Parent(s):
d2cde50
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,7 +32,7 @@ def invert(x0, prompt_src="", num_diffusion_steps=100, cfg_scale_src = 3.5, eta
|
|
| 32 |
|
| 33 |
|
| 34 |
|
| 35 |
-
def sample(
|
| 36 |
|
| 37 |
# reverse process (via Zs and wT)
|
| 38 |
w0, _ = inversion_reverse_process(sd_pipe, xT=wts[skip], etas=eta, prompts=[prompt_tar], cfg_scales=[cfg_scale_tar], prog_bar=True, zs=zs[skip:])
|
|
|
|
| 32 |
|
| 33 |
|
| 34 |
|
| 35 |
+
def sample(zs, wts, prompt_tar="", cfg_scale_tar=15, skip=36, eta = 1):
|
| 36 |
|
| 37 |
# reverse process (via Zs and wT)
|
| 38 |
w0, _ = inversion_reverse_process(sd_pipe, xT=wts[skip], etas=eta, prompts=[prompt_tar], cfg_scales=[cfg_scale_tar], prog_bar=True, zs=zs[skip:])
|