wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-12-23 21:10:49 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Update model_management.py

Browse Source
This commit is contained in:
Vijaysinh 2025-11-28 18:25:40 +05:30 committed by GitHub
parent 0c85c7a92f
commit 387aeed9ce
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
comfy/model_management.py
View File

@ -15,7 +15,8 @@
You should have received a copy of the GNU General Public License You should have received a copy of the GNU General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>. along with this program. If not, see <https://www.gnu.org/licenses/>.
""" """
import gc
import torch
import psutil import psutil
import logging import logging
from enum import Enum from enum import Enum
@ -1524,7 +1525,7 @@ def throw_exception_if_processing_interrupted():
if interrupt_processing: if interrupt_processing:
interrupt_processing = False interrupt_processing = False
raise InterruptProcessingException() raise InterruptProcessingException()
"""
def unload_all_models_full(): def unload_all_models_full():
""" """
Completely unloads all models from RAM and GPU when the user cancels. Completely unloads all models from RAM and GPU when the user cancels.
@ -1549,3 +1550,13 @@ def unload_all_models_full():
logging.info("All models unloaded successfully (manual full unload).") logging.info("All models unloaded successfully (manual full unload).")
except Exception as e: except Exception as e:
logging.warning(f"Model unload warning: {e}") logging.warning(f"Model unload warning: {e}")
"""
def cleanup_ram():
gc.collect()
try:
torch.cuda.empty_cache()
except:
pass