diff --git a/nodes.py b/nodes.py index f069c66..eda4d73 100644 --- a/nodes.py +++ b/nodes.py @@ -99,7 +99,7 @@ class GGUFModelPatcher(comfy.model_patcher.ModelPatcher): if device == self.offload_device: linked.append((n, m)) continue - if linked: + if linked and self.load_device != self.offload_device: logging.info(f"Attempting to release mmap ({len(linked)})") for n, m in linked: # TODO: possible to OOM, find better way to detach