latostadaok commited on
Commit
18e5bb5
·
1 Parent(s): a0466aa

Enable scheduler in DiffusionPipeline and adjust default slider value for steps

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -87,7 +87,7 @@ scheduler_config = {
87
  scheduler = FlowMatchEulerDiscreteScheduler.from_config(scheduler_config)
88
  pipe = DiffusionPipeline.from_pretrained(
89
  base_model,
90
- #scheduler=scheduler,
91
  torch_dtype=dtype
92
  ).to(device)
93
 
@@ -458,7 +458,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, delete_cache=(60, 60)) as app:
458
  minimum=1.0,
459
  maximum=5.0,
460
  step=0.1,
461
- value=3.5,
462
  info="Lower for speed mode, higher for quality"
463
  )
464
  steps = gr.Slider(
 
87
  scheduler = FlowMatchEulerDiscreteScheduler.from_config(scheduler_config)
88
  pipe = DiffusionPipeline.from_pretrained(
89
  base_model,
90
+ scheduler=scheduler,
91
  torch_dtype=dtype
92
  ).to(device)
93
 
 
458
  minimum=1.0,
459
  maximum=5.0,
460
  step=0.1,
461
+ value=1,
462
  info="Lower for speed mode, higher for quality"
463
  )
464
  steps = gr.Slider(