fred-dev commited on
Commit
2d2c267
·
verified ·
1 Parent(s): 5c5ab05

update sigma defaults

Browse files
stable_audio_tools/interface/gradio.py CHANGED
@@ -417,7 +417,7 @@ def create_sampling_ui(model_config):
417
 
418
  with gr.Row():
419
  # Steps slider
420
- steps_slider = gr.Slider(minimum=1, maximum=500, step=1, value=100, label="Steps")
421
 
422
  # Preview Every slider
423
  preview_every_slider = gr.Slider(minimum=0, maximum=100, step=1, value=0, label="Preview Every")
@@ -499,14 +499,14 @@ def create_sampling_ui(model_config):
499
  with gr.Row():
500
  sampler_type_dropdown = gr.Dropdown(["dpmpp-2m-sde", "dpmpp-3m-sde", "k-heun", "k-lms", "k-dpmpp-2s-ancestral", "k-dpm-2", "k-dpm-fast"], label="Sampler type", value="dpmpp-2m-sde")
501
  sigma_min_slider = gr.Slider(minimum=0.0, maximum=2.0, step=0.01, value=0.03, label="Sigma min")
502
- sigma_max_slider = gr.Slider(minimum=0.0, maximum=200.0, step=0.1, value=80, label="Sigma max")
503
- cfg_rescale_slider = gr.Slider(minimum=0.0, maximum=1, step=0.01, value=0.2, label="CFG rescale amount")
504
 
505
 
506
  # Default generation tab
507
  with gr.Accordion("Init audio", open=False):
508
  init_audio_input = gr.Audio(label="Init audio")
509
- init_noise_level_slider = gr.Slider(minimum=0.1, maximum=100.0, step=0.01, value=0.1, label="Init noise level")
510
 
511
  inputs = [
512
  seconds_start_slider,
 
417
 
418
  with gr.Row():
419
  # Steps slider
420
+ steps_slider = gr.Slider(minimum=1, maximum=500, step=1, value=250, label="Steps")
421
 
422
  # Preview Every slider
423
  preview_every_slider = gr.Slider(minimum=0, maximum=100, step=1, value=0, label="Preview Every")
 
499
  with gr.Row():
500
  sampler_type_dropdown = gr.Dropdown(["dpmpp-2m-sde", "dpmpp-3m-sde", "k-heun", "k-lms", "k-dpmpp-2s-ancestral", "k-dpm-2", "k-dpm-fast"], label="Sampler type", value="dpmpp-2m-sde")
501
  sigma_min_slider = gr.Slider(minimum=0.0, maximum=2.0, step=0.01, value=0.03, label="Sigma min")
502
+ sigma_max_slider = gr.Slider(minimum=0.0, maximum=200.0, step=0.1, value=50, label="Sigma max")
503
+ cfg_rescale_slider = gr.Slider(minimum=0.0, maximum=1, step=0.01, value=0.4, label="CFG rescale amount")
504
 
505
 
506
  # Default generation tab
507
  with gr.Accordion("Init audio", open=False):
508
  init_audio_input = gr.Audio(label="Init audio")
509
+ init_noise_level_slider = gr.Slider(minimum=0.1, maximum=100.0, step=0.01, value=1.0, label="Init noise level")
510
 
511
  inputs = [
512
  seconds_start_slider,