diff options
author | AUTOMATIC <16777216c@gmail.com> | 2023-01-14 16:45:39 +0300 |
---|---|---|
committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-14 16:45:39 +0300 |
commit | 69781031e7473e020b3af4461fdceb20130e56ab (patch) | |
tree | ad98db87c0502c63d9214199928463dc194fc1bb /scripts/prompts_from_file.py | |
parent | 27c1b3b816c7f64fae41a7ff0851e3a83ed5499b (diff) |
simplify expression in prompts from file script
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)
|