We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2cf80b9 commit 6feb6d0Copy full SHA for 6feb6d0
paddlenlp/ops/faster_transformer/transformer/decoder.py
@@ -374,7 +374,8 @@ def forward(self, src_word):
374
mem_seq_lens = paddle.sum(paddle.cast(
375
src_word != self.bos_id, dtype="int32"),
376
axis=-1,
377
- keepdim=True)
+ keepdim=True,
378
+ dtype="int32")
379
380
src_slf_attn_bias = paddle.cast(
381
src_word == self.bos_id,
0 commit comments