Skip to content

Commit

Permalink
[docs] correct input documentation for MISTRAL model to reference `in…
Browse files Browse the repository at this point in the history
…put_ids` instead of `decoder_input_ids`
  • Loading branch information
h3110Fr13nd committed Oct 22, 2024
1 parent 9c36adf commit d9c7406
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/transformers/models/mistral/modeling_mistral.py
Original file line number Diff line number Diff line change
Expand Up @@ -619,7 +619,7 @@ def _init_weights(self, module):
Indices can be obtained using [`AutoTokenizer`]. See [`PreTrainedTokenizer.encode`] and
[`PreTrainedTokenizer.__call__`] for details.
If `past_key_values` is used, optionally only the last `decoder_input_ids` have to be input (see
If `past_key_values` is used, optionally only the last `input_ids` have to be input (see
`past_key_values`).
If you want to change padding behavior, you should read [`modeling_opt._prepare_decoder_attention_mask`]
Expand Down

0 comments on commit d9c7406

Please sign in to comment.