diff --git a/nodes.py b/nodes.py index 58934b0..062aafe 100644 --- a/nodes.py +++ b/nodes.py @@ -140,8 +140,7 @@ class GGUFModelPatcher(comfy.model_patcher.ModelPatcher): continue if len(patches) > 0: p.patches = [] - - return super().unpatch_model(device_to=device_to, unpatch_weights=unpatch_weights) + return super().unpatch_model(device_to=None, unpatch_weights=unpatch_weights) def load(self, *args, force_patch_weights=False, **kwargs): # always call `patch_weight_to_device` even for lowvram