[docs] update input documentation for MAMBA2 and MISTRAL models to include cache_position and attention_mask details

[h3110Fr13nd]

What does this PR do?

Mamba2 and Mistral models had Input docstrings not exactly matching with forwrd pass of their respective models.
This doesn't have a big impact.
But I'm trying to add modular models. Which I'm subclassing from Mistral. #33916
And documentation mismatch is somehow not overriding. Which seems to be a lacking in Mistral model.

Fixes # (issue)

Before submitting

• This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
• Did you read the contributor guideline,
  Pull Request section?
• Was this discussed/approved via a Github issue or the forum? Please add a link
  to it if that's the case.
• Did you make sure to update the documentation with your changes? Here are the
  documentation guidelines, and
  here are tips on formatting docstrings.
• Did you write any new necessary tests?

Who can review?

@stevhliu

Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR.

[h3110Fr13nd]

Also, in modelling_mistral.py there's a typo in input docstring

attention_mask
              ...
-            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`).

as suggested by docs of past_key_values and the forward args

past_key_values
            ...
            If `past_key_values` are used, the user can optionally input only the last `input_ids` (those that don't
            have their past key value states given to this model) of shape `(batch_size, 1)` instead of all `input_ids`
            of shape `(batch_size, sequence_length)`.

@stevhliu

[stevhliu]

Thanks for the update :)

If you can rebase on main, that should fix the failing CI tests.

[HuggingFaceDocBuilderDev]

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.