Krebzonide commited on
Commit
58927e7
·
1 Parent(s): fb251c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -12
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
- #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 = """
 
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 = """