Spaces:
Sleeping
Sleeping
fix
Browse files
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 |
-
|
192 |
-
|
193 |
-
|
194 |
-
|
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",
|