.cuda()

scripts/process_utils.py

@@ -222,6 +222,7 @@ def generate_sotai_image(input_image: Image.Image, output_width: int, output_hei
         # EasyNegativeV2の内容
         easy_negative_v2 = "(worst quality, low quality, normal quality:1.4), lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, jpeg artifacts, signature, watermark, username, blurry, artist name, (bad_prompt_version2:0.8)"
         print(f"\ndevice: {sotai_gen_pipe.device}\n")
+        sotai_gen_pipe = sotai_gen_pipe.cuda()
         output = sotai_gen_pipe(
             prompt,
             image=[input_image, input_image],