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