Spaces:
Runtime error
Runtime error
update space
Browse files
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",
|
22 |
)
|
23 |
pipe = pipe.to(device)
|
24 |
|
25 |
MAX_SEED = np.iinfo(np.int32).max
|
26 |
-
MAX_IMAGE_SIZE =
|
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=
|
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=
|
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():
|