Linoy Tsaban
commited on
Commit
•
622ce03
1
Parent(s):
e6519b8
Update app.py
Browse files
app.py
CHANGED
@@ -76,7 +76,6 @@ def edit(input_image,
|
|
76 |
sega_edit_guidance=0,
|
77 |
# warm_up=1,
|
78 |
# neg_guidance=False,
|
79 |
-
flip=False,
|
80 |
left = 0,
|
81 |
right = 0,
|
82 |
top = 0,
|
@@ -89,8 +88,6 @@ def edit(input_image,
|
|
89 |
# invert
|
90 |
# wt, zs, wts = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps, cfg_scale_src=src_cfg_scale)
|
91 |
wt, zs, wts = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps)
|
92 |
-
if flip:
|
93 |
-
wt, zs, wts = torch.flip(wt, [2, 3]),torch.flip(zs, [2, 3]),torch.flip(wts, [2, 3])
|
94 |
|
95 |
latnets = wts[skip].expand(1, -1, -1, -1)
|
96 |
|
@@ -170,7 +167,7 @@ with gr.Blocks() as demo:
|
|
170 |
# reconstruction
|
171 |
skip = gr.Slider(minimum=0, maximum=40, value=36, precision=0, label="Skip Steps", interactive=True)
|
172 |
tar_cfg_scale = gr.Slider(minimum=7, maximum=18,value=15, label=f"Guidance Scale", interactive=True)
|
173 |
-
|
174 |
left = gr.Number(value=0, precision=0, label="Left Shift", interactive=True)
|
175 |
right = gr.Number(value=0, precision=0, label="Right Shift", interactive=True)
|
176 |
top = gr.Number(value=0, precision=0, label="Top Shift", interactive=True)
|
@@ -196,7 +193,6 @@ with gr.Blocks() as demo:
|
|
196 |
sega_edit_guidance,
|
197 |
# warm_up,
|
198 |
# neg_guidance,
|
199 |
-
flip,
|
200 |
left,
|
201 |
right,
|
202 |
top,
|
|
|
76 |
sega_edit_guidance=0,
|
77 |
# warm_up=1,
|
78 |
# neg_guidance=False,
|
|
|
79 |
left = 0,
|
80 |
right = 0,
|
81 |
top = 0,
|
|
|
88 |
# invert
|
89 |
# wt, zs, wts = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps, cfg_scale_src=src_cfg_scale)
|
90 |
wt, zs, wts = invert(x0 =x0 , prompt_src=src_prompt, num_diffusion_steps=steps)
|
|
|
|
|
91 |
|
92 |
latnets = wts[skip].expand(1, -1, -1, -1)
|
93 |
|
|
|
167 |
# reconstruction
|
168 |
skip = gr.Slider(minimum=0, maximum=40, value=36, precision=0, label="Skip Steps", interactive=True)
|
169 |
tar_cfg_scale = gr.Slider(minimum=7, maximum=18,value=15, label=f"Guidance Scale", interactive=True)
|
170 |
+
|
171 |
left = gr.Number(value=0, precision=0, label="Left Shift", interactive=True)
|
172 |
right = gr.Number(value=0, precision=0, label="Right Shift", interactive=True)
|
173 |
top = gr.Number(value=0, precision=0, label="Top Shift", interactive=True)
|
|
|
193 |
sega_edit_guidance,
|
194 |
# warm_up,
|
195 |
# neg_guidance,
|
|
|
196 |
left,
|
197 |
right,
|
198 |
top,
|