Merge branch 'dev' into img2img-batch-png-info

1 files changed, 3 insertions, 7 deletions

diff --git a/scripts/prompts_from_file.py b/scripts/prompts_from_file.py
index b918a764..50320d55 100644
--- a/scripts/prompts_from_file.py
+++ b/scripts/prompts_from_file.py
@@ -1,13 +1,11 @@
 import copy

 import random

-import sys

-import traceback

 import shlex

 

 import modules.scripts as scripts

 import gradio as gr

 

-from modules import sd_samplers

+from modules import sd_samplers, errors

 from modules.processing import Processed, process_images

 from modules.shared import state

 

@@ -123,8 +121,7 @@ class Script(scripts.Script):
         return [checkbox_iterate, checkbox_iterate_batch, prompt_txt]

 

     def run(self, p, checkbox_iterate, checkbox_iterate_batch, prompt_txt: str):

-        lines = [x.strip() for x in prompt_txt.splitlines()]

-        lines = [x for x in lines if len(x) > 0]

+        lines = [x for x in (x.strip() for x in prompt_txt.splitlines()) if x]

 

         p.do_not_save_grid = True

 

@@ -136,8 +133,7 @@ class Script(scripts.Script):
                 try:

                     args = cmdargs(line)

                 except Exception:

-                    print(f"Error parsing line {line} as commandline:", file=sys.stderr)

-                    print(traceback.format_exc(), file=sys.stderr)

+                    errors.report(f"Error parsing line {line} as commandline", exc_info=True)

                     args = {"prompt": line}

             else:

                 args = {"prompt": line}