Fix some issues with the model sometimes not getting patched.

This commit is contained in:
comfyanonymous 2024-08-12 12:27:54 -04:00
parent 446d2ce502
commit c58216564d

View File

@ -669,6 +669,8 @@ class ModelPatcher:
return memory_freed
def partially_load(self, device_to, extra_memory=0):
self.unpatch_model(unpatch_weights=False)
self.patch_model(patch_weights=False)
full_load = False
if self.model.model_lowvram == False:
return 0