diff --git a/onediff_comfy_nodes/utils/model_patcher.py b/onediff_comfy_nodes/utils/model_patcher.py index b012b2b38..15d0937e2 100644 --- a/onediff_comfy_nodes/utils/model_patcher.py +++ b/onediff_comfy_nodes/utils/model_patcher.py @@ -541,6 +541,7 @@ def __init__( else: self.current_device = current_device + self.model_lowvram = getattr(model, "model_lowvram", False) def get_mixed_speedup_class(module_cls): class MixedSpeedUpModelPatcher(OneFlowSpeedUpModelPatcher, module_cls): @@ -584,6 +585,8 @@ def clone(self): n.object_patches = self.object_patches.copy() n.model_options = copy.deepcopy(self.model_options) n.model_keys = self.model_keys + + n.model_lowvram = self.model_lowvram return n return MixedSpeedUpModelPatcher