xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-12 05:33:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix memory lack error

Browse Source
This commit is contained in:
david 2024-08-06 16:33:47 +08:00
parent 66ccde3bd4
commit 03d548fb36
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/cache.py
View File

@ -45,6 +45,8 @@ class ModelCache:
item['sd'] = (keys, values)
def cache_clip(self, key, clip_key, clip):
if not self._cache_state:
return
item = self._get_item(key)
item[clip_key] = clip

4
comfy/model_management.py
View File

@ -503,11 +503,11 @@ def cleanup_models(keep_clone_weights_loaded=False):
to_delete = []
for i in range(len(current_loaded_models)):
if sys.getrefcount(current_loaded_models[i].model) <= 2:
if sys.getrefcount(current_loaded_models[i].model) <= 3:
if not keep_clone_weights_loaded:
to_delete = [i] + to_delete
# TODO: find a less fragile way to do this.
elif sys.getrefcount(current_loaded_models[i].real_model) <= 3: # references from .real_model + the .model
elif sys.getrefcount(current_loaded_models[i].real_model) <= 4: # references from .real_model + the .model
to_delete = [i] + to_delete
for i in to_delete: