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 1e63f61 commit 5260996Copy full SHA for 5260996
paddlenlp/transformers/t5/modeling.py
@@ -697,7 +697,7 @@ def _shift_right(self, input_ids):
697
# replace possible -100 values in labels by `pad_token_id`
698
shifted_input_ids = paddle.where(
699
shifted_input_ids == -100,
700
- paddle.assign(np.asarray(pad_token_id, dtype=data_type_converter(shifted_input_ids))),
+ paddle.assign(np.asarray(pad_token_id, dtype=data_type_converter(shifted_input_ids)).reshape([1])),
701
shifted_input_ids,
702
)
703
0 commit comments