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",
|