Fix some issues with the model sometimes not getting patched.

main
comfyanonymous 6 months ago
parent 517f4a94e4
commit 5d43e75e5b

@ -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

Loading…
Cancel
Save