simplify expression in prompts from file script

3 years ago · 69781031e7
parent 27c1b3b816
commit 69781031e7
1 changed files with 1 additions and 5 deletions
--- a/scripts/prompts_from_file.py
+++ b/scripts/prompts_from_file.py
@ -146,11 +146,7 @@ class Script(scripts.Script):
            else:
                args = {"prompt": line}

-            n_iter = args.get("n_iter", p.n_iter)
-            if n_iter != 1:
-                job_count += n_iter
-            else:
-                job_count += 1
+            job_count += args.get("n_iter", p.n_iter)

            jobs.append(args)