From 9b1f8c8ab532881e889d012ccfcd0b4473dfaecc Mon Sep 17 00:00:00 2001 From: Dinghao Zhou Date: Fri, 23 Feb 2024 19:37:55 +0800 Subject: [PATCH] [transformer] fix bidecoder sdpa --- 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,