Krebzonide commited on
Commit
c8d71a7
·
1 Parent(s): b85fd37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -12
app.py CHANGED
@@ -7,21 +7,24 @@ import gradio as gr
7
 
8
  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
- 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 = """
 
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 = """