From 69781031e7473e020b3af4461fdceb20130e56ab Mon Sep 17 00:00:00 2001 From: AUTOMATIC <16777216c@gmail.com> Date: Sat, 14 Jan 2023 16:45:39 +0300 Subject: [PATCH] simplify expression in prompts from file script --- scripts/prompts_from_file.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/scripts/prompts_from_file.py b/scripts/prompts_from_file.py index 1fe10a7c..f3e711d7 100644 --- 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)