fix to work

kohya-ss · Jun 28, 2023 · 8521ab7 · 8521ab7
1 parent 71a6d49
commit 8521ab7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdxl_minimal_inference.py b/sdxl_minimal_inference.py
@@ -169,7 +169,7 @@ def get_timestep_embedding(x, outdim):
         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 @@ def call_text_encoder(text):
             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: