25 steps works fine

#45
by multimodalart HF staff - opened
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -75,8 +75,8 @@ def generate(
75
  height: int = 1024,
76
  guidance_scale_base: float = 5.0,
77
  guidance_scale_refiner: float = 5.0,
78
- num_inference_steps_base: int = 50,
79
- num_inference_steps_refiner: int = 50,
80
  apply_refiner: bool = False,
81
  ) -> PIL.Image.Image:
82
  generator = torch.Generator().manual_seed(seed)
@@ -209,9 +209,9 @@ with gr.Blocks(css="style.css") as demo:
209
  num_inference_steps_base = gr.Slider(
210
  label="Number of inference steps for base",
211
  minimum=10,
212
- maximum=100,
213
  step=1,
214
- value=50,
215
  )
216
  with gr.Row(visible=False) as refiner_params:
217
  guidance_scale_refiner = gr.Slider(
 
75
  height: int = 1024,
76
  guidance_scale_base: float = 5.0,
77
  guidance_scale_refiner: float = 5.0,
78
+ num_inference_steps_base: int = 25,
79
+ num_inference_steps_refiner: int = 25,
80
  apply_refiner: bool = False,
81
  ) -> PIL.Image.Image:
82
  generator = torch.Generator().manual_seed(seed)
 
209
  num_inference_steps_base = gr.Slider(
210
  label="Number of inference steps for base",
211
  minimum=10,
212
+ maximum=25,
213
  step=1,
214
+ value=25,
215
  )
216
  with gr.Row(visible=False) as refiner_params:
217
  guidance_scale_refiner = gr.Slider(