[transformer] fix bidecoder sdpa (#2368)

wenet-e2e · Feb 26, 2024 · fbbecfd · fbbecfd
1 parent 1a3633a
commit fbbecfd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,