Hugging Face's logo

Spaces:
diffusers
/
benchmark
Paused

App Files Community
patrickvonplaten commited on
Commit
5e0a11c
·
1 Parent(s): 9f27b89

up

Browse files
Files changed (2) hide show
  1. __pycache__/app.cpython-310.pyc +0 -0
  2. app.py +3 -3
__pycache__/app.cpython-310.pyc CHANGED
Binary files a/__pycache__/app.cpython-310.pyc and b/__pycache__/app.cpython-310.pyc differ
 
app.py CHANGED
@@ -8,8 +8,8 @@ USE_TORCH_COMPILE = False
8
  dtype = torch.float16
9
  device = torch.device("cuda:0")
10
 
11
- pipeline = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", variant="fp16", torch_dtype=torch.float16)
12
- pipeline.to("cuda")
13
 
14
 
15
  if USE_TORCH_COMPILE:
@@ -33,7 +33,7 @@ with gr.Blocks(css="style.css") as demo:
33
  minimum=0,
34
  maximum=8,
35
  step=1,
36
- value=0,
37
  )
38
  prompt_len = gr.Slider(
39
  label="Prompt len",
 
8
  dtype = torch.float16
9
  device = torch.device("cuda:0")
10
 
11
+ pipeline = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", variant="fp16", torch_dtype=dtype)
12
+ pipeline.to(device)
13
 
14
 
15
  if USE_TORCH_COMPILE:
 
33
  minimum=0,
34
  maximum=8,
35
  step=1,
36
+ value=1,
37
  )
38
  prompt_len = gr.Slider(
39
  label="Prompt len",