Skip to content

Commit

Permalink
Fix some issues with the model sometimes not getting patched.
Browse files Browse the repository at this point in the history
  • Loading branch information
comfyanonymous committed Aug 12, 2024
1 parent 517f4a9 commit 5d43e75
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions comfy/model_patcher.py
Original file line number Diff line number Diff line change
Expand Up @@ -669,6 +669,8 @@ def partially_unload(self, device_to, memory_to_free=0):
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
Expand Down

0 comments on commit 5d43e75

Please sign in to comment.