transformers
Update Mamba types and pass through use_cache attr to MambaModel
#29605
Merged
Merged
Update Mamba types and pass through use_cache attr to MambaModel #29605
ArthurZucker
merged
14 commits into
huggingface:main
from
koayon:update-mamba-types
Update docstring for RMSNorm
77eacd91
Update cache_params object to correct MambaCache type
c975cf59
Update docstrings and type info
c1211cee
Pass through use_cache
e4aa1b6a
ruff
f9b13ecf
koayon
changed the title 
gante
approved these changes
on 2024-03-14
Reformat with 119 char limit per line (thanks Arthur)
5816a8d5
Pass through use_cache specifically to the backbone rather than all k…
5a3f59d1
koayon
commented
on 2024-03-20
Update src/transformers/models/mamba/modeling_mamba.py
d8aea25d
ArthurZucker
commented
on 2024-03-20
koayon
commented
on 2024-03-20
Update src/transformers/models/mamba/modeling_mamba.py
2c3fa5b8
Update src/transformers/models/mamba/modeling_mamba.py
72f99a01
Update src/transformers/models/mamba/modeling_mamba.py
de6eef1d
Update tab
d16ffcef
koayon
commented
on 2024-03-20
Update src/transformers/models/mamba/modeling_mamba.py
b4438a8e
ArthurZucker
commented
on 2024-03-20
Update src/transformers/models/mamba/modeling_mamba.py
b0e71609
ArthurZucker
approved these changes
on 2024-03-20
Login to write a write a comment.
Login via GitHub