Spaces:
Running
on
A10G
Running
on
A10G
Update app.py
#19
by
phamalor
- opened
app.py
CHANGED
@@ -39,25 +39,25 @@ if SAFETY_CHECKER == "True":
|
|
39 |
i2i_pipe = AutoPipelineForImage2Image.from_pretrained(
|
40 |
"stabilityai/sdxl-turbo",
|
41 |
torch_dtype=torch_dtype,
|
42 |
-
variant="
|
43 |
)
|
44 |
t2i_pipe = AutoPipelineForText2Image.from_pretrained(
|
45 |
"stabilityai/sdxl-turbo",
|
46 |
torch_dtype=torch_dtype,
|
47 |
-
variant="
|
48 |
)
|
49 |
else:
|
50 |
i2i_pipe = AutoPipelineForImage2Image.from_pretrained(
|
51 |
"stabilityai/sdxl-turbo",
|
52 |
safety_checker=None,
|
53 |
torch_dtype=torch_dtype,
|
54 |
-
variant="
|
55 |
)
|
56 |
t2i_pipe = AutoPipelineForText2Image.from_pretrained(
|
57 |
"stabilityai/sdxl-turbo",
|
58 |
safety_checker=None,
|
59 |
torch_dtype=torch_dtype,
|
60 |
-
variant="
|
61 |
)
|
62 |
|
63 |
|
|
|
39 |
i2i_pipe = AutoPipelineForImage2Image.from_pretrained(
|
40 |
"stabilityai/sdxl-turbo",
|
41 |
torch_dtype=torch_dtype,
|
42 |
+
variant="fp12" if torch_dtype == torch.float16 else "fp12",
|
43 |
)
|
44 |
t2i_pipe = AutoPipelineForText2Image.from_pretrained(
|
45 |
"stabilityai/sdxl-turbo",
|
46 |
torch_dtype=torch_dtype,
|
47 |
+
variant="fp12" if torch_dtype == torch.float16 else "fp12",
|
48 |
)
|
49 |
else:
|
50 |
i2i_pipe = AutoPipelineForImage2Image.from_pretrained(
|
51 |
"stabilityai/sdxl-turbo",
|
52 |
safety_checker=None,
|
53 |
torch_dtype=torch_dtype,
|
54 |
+
variant="fp12" if torch_dtype == torch.float16 else "fp12",
|
55 |
)
|
56 |
t2i_pipe = AutoPipelineForText2Image.from_pretrained(
|
57 |
"stabilityai/sdxl-turbo",
|
58 |
safety_checker=None,
|
59 |
torch_dtype=torch_dtype,
|
60 |
+
variant="fp12" if torch_dtype == torch.float16 else "fp12",
|
61 |
)
|
62 |
|
63 |
|