From 8521ab79901608948fd34bcc6a12f850fcb38465 Mon Sep 17 00:00:00 2001
From: Kohya S <ykumeykume@gmail.com>
Date: Wed, 28 Jun 2023 13:09:02 +0900
Subject: [PATCH] fix to work

---
 sdxl_minimal_inference.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/sdxl_minimal_inference.py b/sdxl_minimal_inference.py
index 138d2856..02a4af9c 100644
--- a/sdxl_minimal_inference.py
+++ b/sdxl_minimal_inference.py
@@ -169,7 +169,7 @@ if __name__ == "__main__":
         beta_schedule=SCHEDLER_SCHEDULE,
     )
 
-    def generate_image(text, negative_text, seed=None):
+    def generate_image(prompt, negative_prompt, seed=None):
         # 将来的にサイズ情報も変えられるようにする / Make it possible to change the size information in the future
         # prepare embedding
         with torch.no_grad():
@@ -295,7 +295,7 @@ if __name__ == "__main__":
             img.save(os.path.join(args.output_dir, f"image_{timestamp}_{i:03d}.png"))
 
     if not args.interactive:
-        generate_image(args.prompt, args.negative_prompt, args.seed)
+        generate_image(args.prompt, args.negative_prompt, seed)
     else:
         # loop for interactive
         while True: