sayakpaul HF staff commited on
Commit
5f17264
1 Parent(s): fd14e0a
Files changed (1) hide show
  1. app.py +5 -7
app.py CHANGED
@@ -187,13 +187,11 @@ def generate(
187
  with gr.Blocks() as demo:
188
  do_torch_compile = gr.Checkbox(label="Enable torch.compile()?")
189
  use_channels_last = gr.Checkbox(label="Use `channels_last` memory layout?")
190
- pipeline_to_benchmark = (
191
- gr.Dropdown(
192
- list(pipeline_mapping.keys()),
193
- value=["Stable Diffusion V1.5"],
194
- multiselect=False,
195
- label="Pipeline to benchmark",
196
- ),
197
  )
198
  batch_size = gr.Slider(
199
  label="Number of images per prompt",
 
187
  with gr.Blocks() as demo:
188
  do_torch_compile = gr.Checkbox(label="Enable torch.compile()?")
189
  use_channels_last = gr.Checkbox(label="Use `channels_last` memory layout?")
190
+ pipeline_to_benchmark = gr.Dropdown(
191
+ list(pipeline_mapping.keys()),
192
+ value=["Stable Diffusion V1.5"],
193
+ multiselect=False,
194
+ label="Pipeline to benchmark",
 
 
195
  )
196
  batch_size = gr.Slider(
197
  label="Number of images per prompt",