diff --git a/comfy/model_patcher.py b/comfy/model_patcher.py index 6deb71e120bc..77d2d1b33349 100644 --- a/comfy/model_patcher.py +++ b/comfy/model_patcher.py @@ -732,7 +732,7 @@ def _load_list(self, for_dynamic=False, default_device=None): default = False params = { name: param for name, param in m.named_parameters(recurse=False) } for name, param in m.named_parameters(recurse=True): - if name not in params: + if name not in params and not name.startswith("parametrizations."): default = True # default random weights in non leaf modules break if default and default_device is not None: diff --git a/comfy/sd.py b/comfy/sd.py index e573804a50c2..de2f2e9c91aa 100644 --- a/comfy/sd.py +++ b/comfy/sd.py @@ -596,7 +596,6 @@ def __init__(self, sd=None, device=None, config=None, dtype=None, metadata=None) self.process_output = lambda audio: audio self.process_input = lambda audio: audio self.working_dtypes = [torch.float16, torch.bfloat16, torch.float32] - self.disable_offload = True elif "blocks.2.blocks.3.stack.5.weight" in sd or "decoder.blocks.2.blocks.3.stack.5.weight" in sd or "layers.4.layers.1.attn_block.attn.qkv.weight" in sd or "encoder.layers.4.layers.1.attn_block.attn.qkv.weight" in sd: #genmo mochi vae if "blocks.2.blocks.3.stack.5.weight" in sd: sd = comfy.utils.state_dict_prefix_replace(sd, {"": "decoder."})