Add SDXL

app.py

@@ -1,6 +1,6 @@
 import torch
 import gradio as gr
-from diffusers import StableDiffusion3Pipeline
+from diffusers import DiffusionPipeline
 from utils import (
     attn_maps,
     cross_attn_init,
@@ -13,9 +13,9 @@ from utils import (
 
 cross_attn_init()
 
-pipe = StableDiffusion3Pipeline.from_pretrained(
-    "stabilityai/stable-diffusion-3-medium-diffusers",
-    torch_dtype=torch.bfloat16
+pipe = DiffusionPipeline.from_pretrained(
+    "stabilityai/stable-diffusion-xl-base-1.0",
+    torch_dtype=torch.float16,
 )
 
 pipe = init_pipeline(pipe)