Idor980 commited on
Commit
0e0e5e6
1 Parent(s): 1d05284

update space

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -18,12 +18,12 @@ if torch.cuda.is_available():
18
  pipe = pipe.to(device)
19
  else:
20
  pipe = DiffusionPipeline.from_pretrained(
21
- "stabilityai/sdxl-turbo", torch_dtype=torch.float16, use_safetensors=True
22
  )
23
  pipe = pipe.to(device)
24
 
25
  MAX_SEED = np.iinfo(np.int32).max
26
- MAX_IMAGE_SIZE = 512
27
 
28
 
29
  def infer(
@@ -123,7 +123,7 @@ with gr.Blocks(css=css) as demo:
123
  minimum=256,
124
  maximum=MAX_IMAGE_SIZE,
125
  step=32,
126
- value=512,
127
  )
128
 
129
  height = gr.Slider(
@@ -131,7 +131,7 @@ with gr.Blocks(css=css) as demo:
131
  minimum=256,
132
  maximum=MAX_IMAGE_SIZE,
133
  step=32,
134
- value=512,
135
  )
136
 
137
  with gr.Row():
 
18
  pipe = pipe.to(device)
19
  else:
20
  pipe = DiffusionPipeline.from_pretrained(
21
+ "stabilityai/sdxl-turbo", use_safetensors=True
22
  )
23
  pipe = pipe.to(device)
24
 
25
  MAX_SEED = np.iinfo(np.int32).max
26
+ MAX_IMAGE_SIZE = 256
27
 
28
 
29
  def infer(
 
123
  minimum=256,
124
  maximum=MAX_IMAGE_SIZE,
125
  step=32,
126
+ value=256,
127
  )
128
 
129
  height = gr.Slider(
 
131
  minimum=256,
132
  maximum=MAX_IMAGE_SIZE,
133
  step=32,
134
+ value=256,
135
  )
136
 
137
  with gr.Row():