Update app.py

app.py

@@ -7,21 +7,24 @@ import gradio as gr
 
 model_base = "stabilityai/stable-diffusion-xl-base-1.0"
 
-vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
+#model_url = "https://huggingface.co/Krebzonide/Colossus_Project_XL/blob/main/colossusProjectXLSFW_v202BakedVAE.safetensors"
+model_url = "https://huggingface.co/Krebzonide/Colossus_Project_XL/blob/main/nsfwSevenof9V3_nsfwSevenof9V3.safetensors"
 
-pipe = StableDiffusionXLPipeline.from_pretrained(
-    model_base, vae=vae, torch_dtype=torch.float16, variant="fp16", use_safetensors=True
-)
+vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 
-#pipe = StableDiffusionXLPipeline.from_single_file(
-#    "https://huggingface.co/Krebzonide/Colossus_Project_XL/blob/main/colossusProjectXLSFW_v202BakedVAE.safetensors",
-#    torch_dtype = torch.float16,
-#    variant = "fp16",
-#    vae = vae,
-#    use_safetensors = True,
-#    scheduler_type = "ddim",
-#    use_auth_token="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"
+#pipe = StableDiffusionXLPipeline.from_pretrained(
+#    model_base, vae=vae, torch_dtype=torch.float16, variant="fp16", use_safetensors=True
 #)
+
+pipe = StableDiffusionXLPipeline.from_single_file(
+    model_url,
+    torch_dtype = torch.float16,
+    variant = "fp16",
+    vae = vae,
+    use_safetensors = True,
+    scheduler_type = "ddim",
+    use_auth_token="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"
+)
 pipe = pipe.to("cuda")
 
 css = """