Spaces:
Sleeping
Sleeping
Commit
·
53a3db1
1
Parent(s):
090982e
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,7 +24,7 @@ device = torch.device(
|
|
| 24 |
"cuda" if torch.cuda.is_available() else "xpu" if xpu_available else "cpu"
|
| 25 |
)
|
| 26 |
torch_device = device
|
| 27 |
-
torch_dtype = torch.
|
| 28 |
|
| 29 |
print(f"SAFETY_CHECKER: {SAFETY_CHECKER}")
|
| 30 |
print(f"TORCH_COMPILE: {TORCH_COMPILE}")
|
|
@@ -47,7 +47,8 @@ if SAFETY_CHECKER == "True":
|
|
| 47 |
)
|
| 48 |
t2i_pipe = AutoPipelineForText2Image.from_pretrained(
|
| 49 |
#"stabilityai/sdxl-turbo",
|
| 50 |
-
"wanghuging/demo_model",
|
|
|
|
| 51 |
torch_dtype=torch_dtype,
|
| 52 |
variant="fp16" #if torch_dtype == torch.float16 else "fp32",
|
| 53 |
)
|
|
@@ -60,7 +61,8 @@ else:
|
|
| 60 |
)
|
| 61 |
t2i_pipe = AutoPipelineForText2Image.from_pretrained(
|
| 62 |
#"stabilityai/sdxl-turbo",
|
| 63 |
-
"wanghuging/demo_model",
|
|
|
|
| 64 |
safety_checker=None,
|
| 65 |
torch_dtype=torch_dtype,
|
| 66 |
variant="fp16" #if torch_dtype == torch.float16 else "fp32",
|
|
|
|
| 24 |
"cuda" if torch.cuda.is_available() else "xpu" if xpu_available else "cpu"
|
| 25 |
)
|
| 26 |
torch_device = device
|
| 27 |
+
torch_dtype = torch.float16 #32 #float16
|
| 28 |
|
| 29 |
print(f"SAFETY_CHECKER: {SAFETY_CHECKER}")
|
| 30 |
print(f"TORCH_COMPILE: {TORCH_COMPILE}")
|
|
|
|
| 47 |
)
|
| 48 |
t2i_pipe = AutoPipelineForText2Image.from_pretrained(
|
| 49 |
#"stabilityai/sdxl-turbo",
|
| 50 |
+
# "wanghuging/demo_model",
|
| 51 |
+
"stabilityai/stable-diffusion-xl-base-1.0",
|
| 52 |
torch_dtype=torch_dtype,
|
| 53 |
variant="fp16" #if torch_dtype == torch.float16 else "fp32",
|
| 54 |
)
|
|
|
|
| 61 |
)
|
| 62 |
t2i_pipe = AutoPipelineForText2Image.from_pretrained(
|
| 63 |
#"stabilityai/sdxl-turbo",
|
| 64 |
+
# "wanghuging/demo_model",
|
| 65 |
+
"stabilityai/stable-diffusion-xl-base-1.0",
|
| 66 |
safety_checker=None,
|
| 67 |
torch_dtype=torch_dtype,
|
| 68 |
variant="fp16" #if torch_dtype == torch.float16 else "fp32",
|