From fbbecfdf3fe9298cdc496f8ad4c942a2c868e0bf Mon Sep 17 00:00:00 2001
From: Dinghao Zhou <hamddct@gmail.com>
Date: Mon, 26 Feb 2024 18:36:12 +0800
Subject: [PATCH] [transformer] fix bidecoder sdpa (#2368)

---
 wenet/transformer/decoder.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/wenet/transformer/decoder.py b/wenet/transformer/decoder.py
index ec467ee43..1d7aed1b9 100644
--- a/wenet/transformer/decoder.py
+++ b/wenet/transformer/decoder.py
@@ -301,6 +301,7 @@ def __init__(
     ):
 
         super().__init__()
+        self.use_sdpa = use_sdpa
         self.tie_word_embedding = tie_word_embedding
         self.left_decoder = TransformerDecoder(
             vocab_size,