diff --git a/modules/ui_image_generation.py b/modules/ui_image_generation.py index 6b39c5b5..bcc5bb2d 100644 --- a/modules/ui_image_generation.py +++ b/modules/ui_image_generation.py @@ -848,6 +848,9 @@ def generate(state, save_images=True): if shared.stop_everything: break + if batch_idx > 0: + clear_torch_cache() + generator.manual_seed(int(seed + batch_idx)) # Generate prompt variation if enabled