WanAnimate: Fix regression in image quality when using fp8

Face blocks not doing well in fp8
This commit is contained in:
kijai
2025-09-22 03:12:48 +03:00
parent 1ab9803bda
commit 2c4dfaf4c8
+1 -1
View File
@@ -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