omer11a commited on
Commit
caa1c82
·
1 Parent(s): 4ca09f6

Decreased running time

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -292,9 +292,9 @@ def main():
292
  with gr.Column():
293
  gr.HTML(ADVANCED_OPTION_DESCRIPTION)
294
  batch_size = gr.Slider(minimum=1, maximum=5, step=1, value=1, label="Number of samples (limited to one sample on current space)")
295
- num_guidance_steps = gr.Slider(minimum=5, maximum=20, step=1, value=8, label="Number of timesteps to perform guidance")
296
- init_step_size = gr.Slider(minimum=0, maximum=50, step=0.5, value=25, label="Initial step size")
297
- final_step_size = gr.Slider(minimum=0, maximum=20, step=0.5, value=10, label="Final step size")
298
  num_clusters_per_subject = gr.Slider(minimum=0, maximum=5, step=0.5, value=3, label="Number of clusters per subject")
299
  cross_loss_scale = gr.Slider(minimum=0, maximum=2, step=0.1, value=1, label="Cross-attention loss scale factor")
300
  self_loss_scale = gr.Slider(minimum=0, maximum=2, step=0.1, value=1, label="Self-attention loss scale factor")
 
292
  with gr.Column():
293
  gr.HTML(ADVANCED_OPTION_DESCRIPTION)
294
  batch_size = gr.Slider(minimum=1, maximum=5, step=1, value=1, label="Number of samples (limited to one sample on current space)")
295
+ num_guidance_steps = gr.Slider(minimum=5, maximum=20, step=1, value=5, label="Number of timesteps to perform guidance")
296
+ init_step_size = gr.Slider(minimum=0, maximum=50, step=0.5, value=30, label="Initial step size")
297
+ final_step_size = gr.Slider(minimum=0, maximum=20, step=0.5, value=15, label="Final step size")
298
  num_clusters_per_subject = gr.Slider(minimum=0, maximum=5, step=0.5, value=3, label="Number of clusters per subject")
299
  cross_loss_scale = gr.Slider(minimum=0, maximum=2, step=0.1, value=1, label="Cross-attention loss scale factor")
300
  self_loss_scale = gr.Slider(minimum=0, maximum=2, step=0.1, value=1, label="Self-attention loss scale factor")