Skip to content

Conversation

Josephasafg
Copy link
Owner

@Josephasafg Josephasafg commented Oct 6, 2025

Purpose

This PR enables prefix caching for Mamba1 models.

  • Added support for storing intermediate SSM states at block boundaries

When the PR will be ready, I'll add more tests and detailed descriptions

Test Plan

Most of the tests in test_hybrid.py pass. I am still investigating why some are not passing.

Test Result


Essential Elements of an Effective PR Description Checklist
  • The purpose of the PR, such as "Fix some issue (link existing issues this PR will resolve)".
  • The test plan, such as providing test command.
  • The test results, such as pasting the results comparison before and after, or e2e results
  • (Optional) The necessary documentation update, such as updating supported_models.md and examples for a new model.
  • (Optional) Release notes update. If your change is user facing, please update the release notes draft in the Google Doc.

Signed-off-by: asafg <[email protected]>
Signed-off-by: asafg <[email protected]>
Signed-off-by: asafg <[email protected]>
Signed-off-by: asafg <[email protected]>
Signed-off-by: asafg <[email protected]>
Signed-off-by: asafg <[email protected]>
Signed-off-by: asafg <[email protected]>
Signed-off-by: asafg <[email protected]>
Signed-off-by: asafg <[email protected]>
Signed-off-by: asafg <[email protected]>
Signed-off-by: asafg <[email protected]>
Signed-off-by: asafg <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant