From 7862749cfe1d427a7b81cce92deb8905e99bf96f Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Mon, 4 Aug 2025 01:25:06 +0300 Subject: [PATCH] Fix single text encode --- nodes.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/nodes.py b/nodes.py index 763eed8..3a95256 100644 --- a/nodes.py +++ b/nodes.py @@ -527,8 +527,12 @@ class WanVideoTextEncodeSingle: params_to_keep = {'norm', 'pos_embedding', 'token_embedding'} for name, param in encoder.model.named_parameters(): dtype_to_use = dtype if any(keyword in name for keyword in params_to_keep) else cast_dtype - set_module_tensor_to_device(encoder.model, name, device=device_to, dtype=dtype_to_use, value=encoder.state_dict[name]) - + value = encoder.state_dict[name] if hasattr(encoder, 'state_dict') else encoder.model.state_dict()[name] + set_module_tensor_to_device(encoder.model, name, device=device_to, dtype=dtype_to_use, value=value) + if hasattr(encoder, 'state_dict'): + del encoder.state_dict + mm.soft_empty_cache() + gc.collect() with torch.autocast(device_type=mm.get_autocast_device(device_to), dtype=encoder.dtype, enabled=encoder.quantization != 'disabled'): encoded = encoder([prompt], device_to)