From 69d993f6bbad4976ffbb1a71a9dc9f3d685d6412 Mon Sep 17 00:00:00 2001
From: Lei <45486779+Naylenv@users.noreply.github.com>
Date: Fri, 31 May 2024 11:45:08 +0800
Subject: [PATCH] fix mutual_self_attention.py bug

Fix the bug where the value of `fusion_blocks` is always set to "midup".
---
 src/models/mutual_self_attention.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/models/mutual_self_attention.py b/src/models/mutual_self_attention.py
index a4d5543..5a03cbf 100644
--- a/src/models/mutual_self_attention.py
+++ b/src/models/mutual_self_attention.py
@@ -45,8 +45,8 @@ def __init__(
             style_fidelity,
             reference_attn,
             reference_adain,
-            fusion_blocks,
             batch_size=batch_size,
+            fusion_blocks=fusion_blocks,
         )
 
     def register_reference_hooks(