diff --git a/comfy/model_patcher.py b/comfy/model_patcher.py index 93d19d6fe..972dcf5de 100644 --- a/comfy/model_patcher.py +++ b/comfy/model_patcher.py @@ -736,7 +736,7 @@ class ModelPatcher: 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 e573804a5..de2f2e9c9 100644 --- a/comfy/sd.py +++ b/comfy/sd.py @@ -596,7 +596,6 @@ class VAE: 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."})