Upload genimage.py

genimage.py

@@ -2,12 +2,11 @@ import spaces
 
 
 def load_pipeline():
-    from diffusers import DiffusionPipeline
+    from diffusers import StableDiffusionXLPipeline
     import torch
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-    pipe = DiffusionPipeline.from_pretrained(
+    pipe = StableDiffusionXLPipeline.from_pretrained(
         "John6666/rae-diffusion-xl-v2-sdxl-spo-pcm",
-        custom_pipeline="lpw_stable_diffusion_xl",
         torch_dtype=torch.float16,
     )
     pipe.to(device)