Browse Source

Code cleanups.

pull/546/head
comfyanonymous 1 year ago
parent
commit
30eb92c3cb
  1. 13
      comfy/model_management.py

13
comfy/model_management.py

@ -302,16 +302,15 @@ def unload_model_clones(model):
def free_memory(memory_required, device, keep_loaded=[]):
unloaded_model = False
for i in range(len(current_loaded_models) -1, -1, -1):
if DISABLE_SMART_MEMORY:
current_free_mem = 0
else:
current_free_mem = get_free_memory(device)
if current_free_mem > memory_required:
break
if not DISABLE_SMART_MEMORY:
if get_free_memory(device) > memory_required:
break
shift_model = current_loaded_models[i]
if shift_model.device == device:
if shift_model not in keep_loaded:
current_loaded_models.pop(i).model_unload()
m = current_loaded_models.pop(i)
m.model_unload()
del m
unloaded_model = True
if unloaded_model:

Loading…
Cancel
Save