diff options
Diffstat (limited to 'scripts/prompts_from_file.py')
| -rw-r--r-- | scripts/prompts_from_file.py | 6 |
1 files changed, 1 insertions, 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)
|