Spaces:
Sleeping
Sleeping
Commit
·
dd2e743
1
Parent(s):
8f57828
Update app.py
Browse files
app.py
CHANGED
@@ -44,7 +44,7 @@ if SAFETY_CHECKER == "True":
|
|
44 |
t2i_pipe = AutoPipelineForText2Image.from_pretrained(
|
45 |
"stabilityai/sdxl-turbo",
|
46 |
torch_dtype=torch_dtype,
|
47 |
-
variant="fp16" if torch_dtype == torch.float16 else "fp32",
|
48 |
)
|
49 |
else:
|
50 |
i2i_pipe = AutoPipelineForImage2Image.from_pretrained(
|
@@ -57,7 +57,7 @@ else:
|
|
57 |
"stabilityai/sdxl-turbo",
|
58 |
safety_checker=None,
|
59 |
torch_dtype=torch_dtype,
|
60 |
-
variant="fp16" if torch_dtype == torch.float16 else "fp32",
|
61 |
)
|
62 |
|
63 |
|
@@ -175,27 +175,27 @@ with gr.Blocks(css=css) as demo:
|
|
175 |
step=1,
|
176 |
)
|
177 |
|
178 |
-
with gr.Accordion("Run with diffusers"):
|
179 |
-
|
180 |
-
|
181 |
-
|
182 |
-
|
183 |
-
|
184 |
-
|
185 |
-
|
186 |
-
|
187 |
-
|
188 |
-
|
189 |
-
|
190 |
-
|
191 |
-
|
192 |
-
|
193 |
-
|
194 |
-
|
195 |
-
|
196 |
-
|
197 |
-
|
198 |
-
|
199 |
|
200 |
inputs = [image_input, prompt, strength, steps, seed]
|
201 |
generate_bt.click(fn=predict, inputs=inputs, outputs=image, show_progress=False)
|
|
|
44 |
t2i_pipe = AutoPipelineForText2Image.from_pretrained(
|
45 |
"stabilityai/sdxl-turbo",
|
46 |
torch_dtype=torch_dtype,
|
47 |
+
variant="fp16" #if torch_dtype == torch.float16 else "fp32",
|
48 |
)
|
49 |
else:
|
50 |
i2i_pipe = AutoPipelineForImage2Image.from_pretrained(
|
|
|
57 |
"stabilityai/sdxl-turbo",
|
58 |
safety_checker=None,
|
59 |
torch_dtype=torch_dtype,
|
60 |
+
variant="fp16" #if torch_dtype == torch.float16 else "fp32",
|
61 |
)
|
62 |
|
63 |
|
|
|
175 |
step=1,
|
176 |
)
|
177 |
|
178 |
+
# with gr.Accordion("Run with diffusers"):
|
179 |
+
# gr.Markdown(
|
180 |
+
# """## Running SDXL Turbo with `diffusers`
|
181 |
+
# ```bash
|
182 |
+
# pip install diffusers==0.23.1
|
183 |
+
# ```
|
184 |
+
# ```py
|
185 |
+
# from diffusers import DiffusionPipeline
|
186 |
+
# pipe = DiffusionPipeline.from_pretrained(
|
187 |
+
# "stabilityai/sdxl-turbo"
|
188 |
+
# ).to("cuda")
|
189 |
+
# results = pipe(
|
190 |
+
# prompt="A cinematic shot of a baby racoon wearing an intricate italian priest robe",
|
191 |
+
# num_inference_steps=1,
|
192 |
+
# guidance_scale=0.0,
|
193 |
+
# )
|
194 |
+
# imga = results.images[0]
|
195 |
+
# imga.save("image.png")
|
196 |
+
# ```
|
197 |
+
# """
|
198 |
+
# )
|
199 |
|
200 |
inputs = [image_input, prompt, strength, steps, seed]
|
201 |
generate_bt.click(fn=predict, inputs=inputs, outputs=image, show_progress=False)
|