Spaces:
Running
on
A10G
Running
on
A10G
fix: url.
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ dtype = torch.float16
|
|
22 |
device = torch.device("cuda")
|
23 |
|
24 |
# pipeline_to_benchmark, batch_size, use_channels_last, do_torch_compile
|
25 |
-
examples = [["SD T2I", 4, True, True]]
|
26 |
|
27 |
pipeline_mapping = {
|
28 |
"SD T2I": (DiffusionPipeline, "runwayml/stable-diffusion-v1-5"),
|
@@ -235,13 +235,13 @@ with gr.Blocks(css="style.css") as demo:
|
|
235 |
)
|
236 |
result = gr.Text(label="Result")
|
237 |
|
238 |
-
gr.Examples(
|
239 |
-
|
240 |
-
|
241 |
-
|
242 |
-
|
243 |
-
|
244 |
-
)
|
245 |
|
246 |
btn.click(
|
247 |
fn=generate,
|
|
|
22 |
device = torch.device("cuda")
|
23 |
|
24 |
# pipeline_to_benchmark, batch_size, use_channels_last, do_torch_compile
|
25 |
+
# examples = [["SD T2I", 4, True, True]]
|
26 |
|
27 |
pipeline_mapping = {
|
28 |
"SD T2I": (DiffusionPipeline, "runwayml/stable-diffusion-v1-5"),
|
|
|
235 |
)
|
236 |
result = gr.Text(label="Result")
|
237 |
|
238 |
+
# gr.Examples(
|
239 |
+
# examples=examples,
|
240 |
+
# inputs=[pipeline_to_benchmark, batch_size, use_channels_last, do_torch_compile],
|
241 |
+
# outputs=result,
|
242 |
+
# fn=generate,
|
243 |
+
# cache_examples=True,
|
244 |
+
# )
|
245 |
|
246 |
btn.click(
|
247 |
fn=generate,
|