sayakpaul HF staff commited on
Commit
1b5a906
1 Parent(s): cf42f63
Files changed (1) hide show
  1. app.py +8 -8
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
- 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,
 
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,