From 4eabdcce7898ff89afb554e477baa20bcc69291f Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 30 Jul 2025 18:45:19 +0300
Subject: [PATCH] wrong way around

---
 nodes_model_loading.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/nodes_model_loading.py b/nodes_model_loading.py
index 2bce948..aadd5f0 100644
--- a/nodes_model_loading.py
+++ b/nodes_model_loading.py
@@ -1047,10 +1047,10 @@ class WanVideoModelLoader:
                         leave=True):
                     dtype_to_use = base_dtype if any(keyword in name for keyword in params_to_keep) else dtype
                     dtype_to_use = dtype if sd[name].dtype == dtype else dtype_to_use
-                    if "patch_embedding" in name:
-                        dtype_to_use = torch.float32
                     if "modulation" in name or "norm" in name or "bias" in name:
                         dtype_to_use = base_dtype
+                    if "patch_embedding" in name:
+                        dtype_to_use = torch.float32
                     set_module_tensor_to_device(transformer, name, device=transformer_load_device, dtype=dtype_to_use, value=sd[name])
                     pbar.update(1)