Upload genimage.py

genimage.py

@@ -7,7 +7,8 @@ def load_pipeline():
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
     pipe = AutoPipelineForText2Image.from_pretrained(
         "John6666/rae-diffusion-xl-v2-sdxl-spo-pcm",
-        custom_pipeline="lpw_stable_diffusion_xl",
+        #custom_pipeline="lpw_stable_diffusion_xl",
+        custom_pipeline="nyanko7/sdxl_smoothed_energy_guidance",
         torch_dtype=torch.float16,
     )
     pipe.to(device)
@@ -35,11 +36,11 @@ pipe = load_pipeline()
 @spaces.GPU
 def generate_image(prompt, neg_prompt):
     metadata = {
-        "prompt": prompt,
-        "negative_prompt": neg_prompt,
+        "prompt": prompt + ", anime, masterpiece, best quality, very aesthetic, absurdres",
+        "negative_prompt": neg_prompt + ", bad hands, bad feet, lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract], photo, deformed, disfigured, low contrast, photo, deformed, disfigured, low contrast",
         "resolution": f"{1024} x {1024}",
-        "guidance_scale": 7.5,
-        "num_inference_steps": 24,
+        "guidance_scale": 7.0,
+        "num_inference_steps": 28,
         "sampler": "Euler",
     }
     try: 
@@ -48,10 +49,10 @@ def generate_image(prompt, neg_prompt):
             negative_prompt=neg_prompt + ", bad hands, bad feet, lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract], photo, deformed, disfigured, low contrast, photo, deformed, disfigured, low contrast",
             width=1024,
             height=1024,
-            guidance_scale=7.5,
-            num_inference_steps=24,
+            guidance_scale=7.0, seg_scale=3.0, seg_applied_layers=["mid"],
+            num_inference_steps=28,
             output_type="pil",
-            clip_skip=1,
+            #clip_skip=1,
         ).images
         if images:
             image_paths = [
@@ -60,5 +61,6 @@ def generate_image(prompt, neg_prompt):
             ]
         return image_paths
     except Exception as e:
+        print(e)
         return []