blanchon commited on
Commit
fd3a897
·
verified ·
1 Parent(s): 73fa763

Change num_inference_steps

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,7 +38,7 @@ def predict(
38
  room_image_category: str,
39
  custom_prompt: str | None = None,
40
  seed: int = 0,
41
- num_inference_steps: int = 28,
42
  max_dimension: int = 1024,
43
  condition_scale: float = 1.0,
44
  progress: gr.Progress = gr.Progress(track_tqdm=True), # noqa: ARG001, B008
@@ -171,7 +171,7 @@ with gr.Blocks(css=css) as demo:
171
  minimum=1,
172
  maximum=50,
173
  step=1,
174
- value=28,
175
  )
176
 
177
  run_button.click(
 
38
  room_image_category: str,
39
  custom_prompt: str | None = None,
40
  seed: int = 0,
41
+ num_inference_steps: int = 4,
42
  max_dimension: int = 1024,
43
  condition_scale: float = 1.0,
44
  progress: gr.Progress = gr.Progress(track_tqdm=True), # noqa: ARG001, B008
 
171
  minimum=1,
172
  maximum=50,
173
  step=1,
174
+ value=4,
175
  )
176
 
177
  run_button.click(