From 97954cd06efc755ddf14e1edb553f93c0921cdda Mon Sep 17 00:00:00 2001 From: Ethanfel Date: Tue, 10 Feb 2026 18:18:34 +0100 Subject: [PATCH] Fix shared tensor warning by cloning before save Aliased tensors (patch_embedding / expanded_patch_embedding / original_patch_embedding) share memory, causing safetensors save_file to warn about duplicates. Clone each tensor to break sharing while keeping all keys for loader compatibility. Co-Authored-By: Claude Opus 4.6 --- nodes.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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}")