aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-01-14 16:45:39 +0300
committerAUTOMATIC <16777216c@gmail.com>2023-01-14 16:45:39 +0300
commit69781031e7473e020b3af4461fdceb20130e56ab (patch)
treead98db87c0502c63d9214199928463dc194fc1bb
parent27c1b3b816c7f64fae41a7ff0851e3a83ed5499b (diff)
simplify expression in prompts from file script
-rw-r--r--scripts/prompts_from_file.py6
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)