linoyts HF staff commited on
Commit
a3ae4e4
·
verified ·
1 Parent(s): 18180a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +23 -8
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
- num_inference_steps = gr.Slider(
159
- label="num inference steps",
160
- minimum=1,
161
- maximum=50,
162
- step=1,
163
- value=8,
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
  )