Upload genimage.py

genimage.py

@@ -6,7 +6,7 @@ def load_pipeline():
     import torch
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
     pipe = StableDiffusionXLPipeline.from_pretrained(
-        "John6666/mala-anime-mix-nsfw-pony-xl-v6-sdxl-spo-pcm",
+        "John6666/rae-diffusion-xl-v2-sdxl-spo-pcm",
         torch_dtype=torch.float16,
     )
     pipe.to(device)
@@ -38,7 +38,7 @@ def generate_image(prompt, neg_prompt):
         "negative_prompt": neg_prompt,
         "resolution": f"{1024} x {1024}",
         "guidance_scale": 7.5,
-        "num_inference_steps": 28,
+        "num_inference_steps": 16,
         "sampler": "Euler",
     }
     try: 
@@ -48,7 +48,7 @@ def generate_image(prompt, neg_prompt):
             width=1024,
             height=1024,
             guidance_scale=7.5,
-            num_inference_steps=28,
+            num_inference_steps=16,
             output_type="pil",
             clip_skip=1,
         ).images