diff --git a/nodes.py b/nodes.py index d0a93fe..5902bc2 100644 --- a/nodes.py +++ b/nodes.py @@ -86,7 +86,11 @@ class WanVideoSaveMergedModel: tensor = v.cpu() if target_dtype is not None: tensor = tensor.to(target_dtype) - clean_sd[k] = tensor + # Clone to break shared memory between aliased tensors + # (e.g. patch_embedding / expanded_patch_embedding / original_patch_embedding) + # safetensors save_file doesn't handle shared tensors, and save_model + # deduplicates keys which breaks compatibility with ComfyUI's load_file + clean_sd[k] = tensor.clone() pbar.update(1) log.info(f"Saving merged WanVideo model to: {output_path}")