oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-01-18 14:40:53 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Clear the torch cache between sequential image generations

Browse source
This commit is contained in:
oobabooga 2025-12-07 11:49:22 -08:00
parent 3b8369a679
commit 1ec9f708e5
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
modules/ui_image_generation.py
View file

@ -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