Skip to content

Commit a8e5419

Browse files
deepllzlixcli
authored andcommitted
Fix qwen model use_casual_mask (PaddlePaddle#9009)
1 parent eddc061 commit a8e5419

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

paddlenlp/transformers/qwen/modeling.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -811,7 +811,7 @@ def forward(
811811
hidden_states = inputs_embeds
812812
use_casual_mask = get_use_casual_mask()
813813
# bool 4D mask
814-
if use_casual_mask is None:
814+
if use_casual_mask is True:
815815
attention_mask = None
816816
else:
817817
attention_mask = self.get_masks(input_shape[0], input_shape[1], past_length, padding_mask=attention_mask)

0 commit comments

Comments
 (0)