Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -54,6 +54,8 @@ def invert_and_edit(image,
|
|
54 |
do_inversion,
|
55 |
seed,
|
56 |
randomize_seed,
|
|
|
|
|
57 |
):
|
58 |
if randomize_seed:
|
59 |
seed = random.randint(0, MAX_SEED)
|
@@ -69,7 +71,9 @@ def invert_and_edit(image,
|
|
69 |
start_timestep = start_timestep/num_inference_steps,
|
70 |
stop_timestep = stop_timestep/num_inference_steps,
|
71 |
num_inference_steps = num_inference_steps,
|
72 |
-
eta=eta,
|
|
|
|
|
73 |
).images[0]
|
74 |
|
75 |
return output, inverted_latents.cpu(), image_latents.cpu(), latent_image_ids.cpu(), do_inversion, seed
|
@@ -155,13 +159,22 @@ based on the implementations of [@raven38](https://github.com/raven38) & [@DarkM
|
|
155 |
)
|
156 |
|
157 |
randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
|
158 |
-
|
159 |
-
|
160 |
-
|
161 |
-
|
162 |
-
|
163 |
-
|
164 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
165 |
|
166 |
with gr.Row():
|
167 |
gamma = gr.Slider(
|
@@ -218,6 +231,8 @@ based on the implementations of [@raven38](https://github.com/raven38) & [@DarkM
|
|
218 |
do_inversion,
|
219 |
seed,
|
220 |
randomize_seed,
|
|
|
|
|
221 |
],
|
222 |
outputs=[result, inverted_latents, image_latents, latent_image_ids, do_inversion, seed],
|
223 |
)
|
|
|
54 |
do_inversion,
|
55 |
seed,
|
56 |
randomize_seed,
|
57 |
+
eta_decay,
|
58 |
+
decay_power,
|
59 |
):
|
60 |
if randomize_seed:
|
61 |
seed = random.randint(0, MAX_SEED)
|
|
|
71 |
start_timestep = start_timestep/num_inference_steps,
|
72 |
stop_timestep = stop_timestep/num_inference_steps,
|
73 |
num_inference_steps = num_inference_steps,
|
74 |
+
eta=eta,
|
75 |
+
decay_eta = eta_decay,
|
76 |
+
eta_decay_power = decay_power,
|
77 |
).images[0]
|
78 |
|
79 |
return output, inverted_latents.cpu(), image_latents.cpu(), latent_image_ids.cpu(), do_inversion, seed
|
|
|
159 |
)
|
160 |
|
161 |
randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
|
162 |
+
with gr.Row():
|
163 |
+
num_inference_steps = gr.Slider(
|
164 |
+
label="num inference steps",
|
165 |
+
minimum=1,
|
166 |
+
maximum=50,
|
167 |
+
step=1,
|
168 |
+
value=8,
|
169 |
+
)
|
170 |
+
eta_decay = gr.Checkbox(label="eta decay", value=False)
|
171 |
+
decay_power = gr.Slider(
|
172 |
+
label="eta decay power",
|
173 |
+
minimum=0,
|
174 |
+
maximum=5,
|
175 |
+
step=1,
|
176 |
+
value=1,
|
177 |
+
)
|
178 |
|
179 |
with gr.Row():
|
180 |
gamma = gr.Slider(
|
|
|
231 |
do_inversion,
|
232 |
seed,
|
233 |
randomize_seed,
|
234 |
+
eta_decay,
|
235 |
+
decay_power,
|
236 |
],
|
237 |
outputs=[result, inverted_latents, image_latents, latent_image_ids, do_inversion, seed],
|
238 |
)
|