Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
Mddct authored Feb 7, 2025
1 parent 3187c98 commit 6533f8f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions wenet/transformer/decoder.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit 6533f8f

Please sign in to comment.