Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

How to use Mamba2 like a RNN model? #622

Open
YuBicheng opened this issue Nov 15, 2024 · 3 comments
Open

How to use Mamba2 like a RNN model? #622

YuBicheng opened this issue Nov 15, 2024 · 3 comments

Comments

@YuBicheng
Copy link

YuBicheng commented Nov 15, 2024 •
edited
Loading

In my task, the length of the input sequence is very long. During training, is it possible to get the hidden state of Mamba2, use it as a feature, and then feed the remaining input and hidden features into Mamba2 and continue the operation? Just like RNN
@YuBicheng YuBicheng changed the title How to use Mamba2 like a RNN How to use Mamba2 like a RNN model? Nov 15, 2024
@jialiangZ
Copy link

I have the same idea. Have you solved it?

1 similar comment
@chaserd
Copy link

chaserd commented Dec 15, 2024

I have the same idea. Have you solved it?

@jialiangZ
Copy link

这个仓库实现了https://github.com/tommyip/mamba2-minimal/blob/4f4413731f24303b4e046962bce6ca5b6a3b13d6/mamba2.py#L199

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@YuBicheng @chaserd @jialiangZ