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

Clear the torch cache before generating

Browse source
This commit is contained in:
oobabooga 2025-12-02 06:41:08 -08:00
parent f4b27a39af
commit e7a15a0361
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
modules/ui_image_generation.py
View file

@ -667,6 +667,10 @@ def generate(state):
"""
import torch
from modules.torch_utils import clear_torch_cache
clear_torch_cache()
try:
model_name = state['image_model_menu']