Spaces:
Runtime error
Runtime error
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 |
)
|