Update Mamba types and pass through use_cache attr to MambaModel (#29605)
* Update docstring for RMSNorm
* Update cache_params object to correct MambaCache type
* Update docstrings and type info
* Pass through use_cache
* ruff
* Reformat with 119 char limit per line (thanks Arthur)
* Pass through use_cache specifically to the backbone rather than all keyword arguments
* Update src/transformers/models/mamba/modeling_mamba.py
* Update src/transformers/models/mamba/modeling_mamba.py
* Update src/transformers/models/mamba/modeling_mamba.py
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
* Update src/transformers/models/mamba/modeling_mamba.py
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
* Update tab
* Update src/transformers/models/mamba/modeling_mamba.py
* Update src/transformers/models/mamba/modeling_mamba.py
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
---------
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>