Update app.py

app.py

@@ -8,8 +8,18 @@ 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)
-pipe = StableDiffusionXLPipeline.from_pretrained(
-    model_base, vae=vae, torch_dtype=torch.float16, variant="fp16", use_safetensors=True
+
+#pipe = StableDiffusionXLPipeline.from_pretrained(
+#    model_base, vae=vae, torch_dtype=torch.float16, variant="fp16", use_safetensors=True
+#)
+
+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"
 )
 pipe = pipe.to("cuda")