Krebzonide
commited on
Commit
·
58927e7
1
Parent(s):
fb251c5
Update app.py
Browse files
app.py
CHANGED
@@ -9,19 +9,19 @@ model_base = "stabilityai/stable-diffusion-xl-base-1.0"
|
|
9 |
|
10 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
11 |
|
12 |
-
|
13 |
-
|
14 |
-
#)
|
15 |
-
|
16 |
-
pipe = StableDiffusionXLPipeline.from_single_file(
|
17 |
-
"https://huggingface.co/Krebzonide/Colossus_Project_XL/blob/main/colossusProjectXLSFW_v202BakedVAE.safetensors",
|
18 |
-
torch_dtype = torch.float16,
|
19 |
-
variant = "fp16",
|
20 |
-
vae = vae,
|
21 |
-
use_safetensors = True,
|
22 |
-
scheduler_type = "ddim",
|
23 |
-
use_auth_token="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"
|
24 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
25 |
pipe = pipe.to("cuda")
|
26 |
|
27 |
css = """
|
|
|
9 |
|
10 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
11 |
|
12 |
+
pipe = StableDiffusionXLPipeline.from_pretrained(
|
13 |
+
model_base, vae=vae, torch_dtype=torch.float16, variant="fp16", use_safetensors=True
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
14 |
)
|
15 |
+
|
16 |
+
#pipe = StableDiffusionXLPipeline.from_single_file(
|
17 |
+
# "https://huggingface.co/Krebzonide/Colossus_Project_XL/blob/main/colossusProjectXLSFW_v202BakedVAE.safetensors",
|
18 |
+
# torch_dtype = torch.float16,
|
19 |
+
# variant = "fp16",
|
20 |
+
# vae = vae,
|
21 |
+
# use_safetensors = True,
|
22 |
+
# scheduler_type = "ddim",
|
23 |
+
# use_auth_token="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"
|
24 |
+
#)
|
25 |
pipe = pipe.to("cuda")
|
26 |
|
27 |
css = """
|