Upload 2 files

app.py

@@ -104,8 +104,8 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=css, delete_ca
         queue=True,
         show_progress="full",
         trigger_mode="once",
-    ).success(dolphin_parse, [chatbot, state], [output_text, copy_btn, copy_btn_pony]).success(
-        convert_danbooru_to_e621_prompt, [output_text, tag_type], [output_text_pony], queue=False,
+    ).success(dolphin_parse, [chatbot, state], [output_text, copy_btn, copy_btn_pony]
+    ).success(convert_danbooru_to_e621_prompt, [output_text, tag_type], [output_text_pony], queue=False,
     ).success(insert_recom_prompt, [output_text, dummy_np, recom_animagine], [output_text, dummy_np], queue=False,
     ).success(insert_recom_prompt, [output_text_pony, dummy_np_pony, recom_pony], [output_text_pony, dummy_np_pony], queue=False)
     chat_clear.click(lambda: None, None, chatbot, queue=False)

genimage.py

@@ -2,16 +2,18 @@ import spaces
 import torch
 
 
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+
+
 def load_pipeline():
     from diffusers import DiffusionPipeline
-    device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
     pipe = DiffusionPipeline.from_pretrained(
         "John6666/rae-diffusion-xl-v2-sdxl-spo-pcm",
         custom_pipeline="lpw_stable_diffusion_xl",
         #custom_pipeline="nyanko7/sdxl_smoothed_energy_guidance",
         torch_dtype=torch.float16,
     )
-    pipe.to(device)
+    pipe.to("cpu")
     return pipe
 
 
@@ -65,6 +67,7 @@ pipe = load_pipeline()
 @torch.inference_mode()
 @spaces.GPU
 def generate_image(prompt, neg_prompt):
+    pipe.to(device)
     prompt += ", anime, masterpiece, best quality, very aesthetic, absurdres"
     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"
     metadata = {
@@ -85,7 +88,7 @@ def generate_image(prompt, neg_prompt):
             guidance_scale=7.0,# seg_scale=3.0, seg_applied_layers=["mid"],
             num_inference_steps=28,
             output_type="pil",
-            clip_skip=1,
+            clip_skip=2,
         ).images
         if images:
             image_paths = [