From 6533f8fd6c109cd7d7d04e8ba0c5b7a4d470404d Mon Sep 17 00:00:00 2001 From: Dinghao Zhou Date: Fri, 7 Feb 2025 12:39:57 +0800 Subject: [PATCH] fix lint --- wenet/transformer/decoder.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/wenet/transformer/decoder.py b/wenet/transformer/decoder.py index 591c19aee..e23542050 100644 --- a/wenet/transformer/decoder.py +++ b/wenet/transformer/decoder.py @@ -126,8 +126,8 @@ def __init__( value_bias, use_sdpa, n_kv_head, head_dim), WENET_ATTENTION_CLASSES["crossattn"]( attention_heads, attention_dim, src_attention_dropout_rate, - src_query_bias, src_key_bias, src_value_bias, use_sdpa, n_kv_head, - head_dim) if src_attention else None, + src_query_bias, src_key_bias, src_value_bias, use_sdpa, + n_kv_head, head_dim) if src_attention else None, mlp_class(attention_dim, linear_units, dropout_rate,