Skip to content

Commit fe42fea

Browse files
authored
Del self in GenerationMixin. (#746)
Del self in GenerationMixin.
1 parent 3dd0bd4 commit fe42fea

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddlenlp/transformers/generation_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -287,7 +287,7 @@ def prepare_attention_mask_for_generation(input_ids, pad_token_id,
287287
is_pad_token_not_equal_to_eos_token_id = (eos_token_id is None) or (
288288
(eos_token_id is not None) and (pad_token_id != eos_token_id))
289289
if is_pad_token_in_inputs_ids and is_pad_token_not_equal_to_eos_token_id:
290-
attention_mask = (input_ids == self.pad_token_id
290+
attention_mask = (input_ids == pad_token_id
291291
).astype(paddle.get_default_dtype()) * -1e9
292292
else:
293293
attention_mask = paddle.zeros_like(

0 commit comments

Comments
 (0)