wrong way around

2025-07-30 18:45:19 +03:00
parent 138b377d1d
commit 4eabdcce78
1 changed files with 2 additions and 2 deletions
@@ -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)