PR #30023 Add docstrings and types for MambaCache

Add docstrings and types for MambaCache #30023

ArthurZucker merged 8 commits into huggingface:main from koayon:add-docstrings-for-mamba-cache

koayon commented on 2024-04-03

ArthurZucker approved these changes on 2024-04-05

Add docstrings and types for MambaCache

e250377f

Update src/transformers/models/mamba/modeling_mamba.py

7e76a922

Update src/transformers/models/mamba/modeling_mamba.py

68f9cc83

Update src/transformers/models/mamba/modeling_mamba.py

329d1f4c

make fixup

2bf4fd2d

import copy in generation_whisper

7a557888

ruff

06bfa6ad

Revert "make fixup"

55731500

koayon force pushed to 55731500 1 year ago

ArthurZucker merged d9fa13ce into main 1 year ago

Reviewers

ArthurZucker

Assignees

No one assigned

Labels

None yet

Milestone

No milestone