Skip to content

Commit c05a08a

Browse files
Add back function. (#12234)
1 parent de9ada6 commit c05a08a

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

comfy/model_management.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -767,6 +767,9 @@ def load_models_gpu(models, memory_required=0, force_patch_weights=False, minimu
767767
load_models_gpu_orig(models, memory_required=memory_required, force_patch_weights=force_patch_weights,
768768
minimum_memory_required=minimum_memory_required, force_full_load=force_full_load)
769769

770+
def load_model_gpu(model):
771+
return load_models_gpu([model])
772+
770773
def loaded_models(only_currently_used=False):
771774
output = []
772775
for m in current_loaded_models:

0 commit comments

Comments
 (0)