diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py
index 7040120..b8a802f 100644
--- a/wanvideo/modules/model.py
+++ b/wanvideo/modules/model.py
@@ -1019,7 +1019,7 @@ class WanModel(ModelMixin, ConfigMixin):
             # blocks
             if model_type == 't2v':
                 cross_attn_type = 't2v_cross_attn'
-            elif model_type == 'i2v':
+            elif model_type == 'i2v' or model_type == 'fl2v':
                 cross_attn_type = 'i2v_cross_attn'
             else:
                 cross_attn_type = 'no_cross_attn'