compatible fix for #19

Kahsolt · Mar 30, 2023 · cdac6a2 · cdac6a2
1 parent 2a0b796
commit cdac6a2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/prompt_travel.py b/scripts/prompt_travel.py
@@ -214,7 +214,7 @@ def get_conds_with_caching(function, required_prompts, steps, cache):
         seeds = p.all_seeds[n * p.batch_size:(n + 1) * p.batch_size]
         subseeds = p.all_subseeds[n * p.batch_size:(n + 1) * p.batch_size]
 
-        if p.scripts is not None:
+        if p.scripts is not None and hasattr(p, 'before_process_batch'):
             p.scripts.before_process_batch(p, batch_number=n, prompts=prompts, seeds=seeds, subseeds=subseeds)
 
         if len(prompts) == 0: