From 2c4dfaf4c82142e72c5f855032779cbd772b2471 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Mon, 22 Sep 2025 03:12:48 +0300
Subject: [PATCH] WanAnimate: Fix regression in image quality when using fp8

Face blocks not doing well in fp8
---
 nodes_model_loading.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nodes_model_loading.py b/nodes_model_loading.py
index 9138f75..46128a6 100644
--- a/nodes_model_loading.py
+++ b/nodes_model_loading.py
@@ -773,7 +773,7 @@ def rename_fuser_block(name):
 def load_weights(transformer, sd=None, weight_dtype=None, base_dtype=None, 
                  transformer_load_device=None, block_swap_args=None, gguf=False, reader=None, patcher=None):
     params_to_keep = {"time_in", "patch_embedding", "time_", "modulation", "text_embedding", 
-                      "adapter", "add", "ref_conv", "casual_audio_encoder", "cond_encoder", "frame_packer", "audio_proj_glob", "face_encoder"}
+                      "adapter", "add", "ref_conv", "casual_audio_encoder", "cond_encoder", "frame_packer", "audio_proj_glob", "face_encoder", "fuser_block"}
     param_count = sum(1 for _ in transformer.named_parameters())
     pbar = ProgressBar(param_count)
     cnt = 0