Skip to content

Conversation

@Maxusmusti
Copy link
Contributor

@Maxusmusti Maxusmusti commented Oct 8, 2025

Updating transformers minimum version to 4.57.0 for Granite 4 support, as well as including mamba-ssm[causal-conv1d]>=2.2.5 for efficient Mamba model support (Granite 4 included).

Summary by CodeRabbit

  • New Features

    • Added optional CUDA acceleration via the mamba-ssm package (causal-conv1d), enabling faster inference and training on supported GPUs. Users can opt in by installing the CUDA extras.
  • Chores

    • Updated the Transformers dependency to >=4.57.0 to improve compatibility, stability, and access to newer model features and fixes.

@coderabbitai
Copy link

coderabbitai bot commented Oct 8, 2025

Walkthrough

Updates dependency constraints in pyproject.toml: raises the minimum transformers version to >=4.57.0 and adds an optional CUDA extra dependency mamba-ssm[causal-conv1d]>=2.2.5.

Changes

Cohort / File(s) Summary
Dependency constraints
pyproject.toml
- Bump: transformers >=4.55.0 → >=4.57.0
- Add CUDA extra: mamba-ssm[causal-conv1d]>=2.2.5

Estimated code review effort

🎯 1 (Trivial) | ⏱️ ~3 minutes

Poem

I twitch my whiskers at version signs,
New gears click in the dependency lines.
Transformers stride to fifty-seven’s gate,
Mamba coils with CUDA’s fate.
I thump approval—light and swift—
Hop-ready builds, a tidy lift.

Pre-merge checks and finishing touches

✅ Passed checks (3 passed)
Check name Status Explanation
Description Check ✅ Passed Check skipped - CodeRabbit’s high-level summary is enabled.
Title Check ✅ Passed The title clearly summarizes the primary change of updating minimum dependencies to support Granite 4 and Mamba, matching the core scope of the pull request without extraneous detail.
Docstring Coverage ✅ Passed No functions found in the changes. Docstring coverage check skipped.
✨ Finishing touches
🧪 Generate unit tests (beta)
  • Create PR with unit tests
  • Post copyable unit tests in a comment
  • Commit unit tests in branch mamba-update

📜 Recent review details

Configuration used: CodeRabbit UI

Review profile: CHILL

Plan: Pro

📥 Commits

Reviewing files that changed from the base of the PR and between fc2175d and c560d29.

📒 Files selected for processing (1)
  • pyproject.toml (2 hunks)
🔇 Additional comments (1)
pyproject.toml (1)

20-62: Confirm PyPI package versions and extras availability

Manually verify that transformers 4.57.0 is published and mamba-ssm 2.2.5 defines a causal-conv1d extra; missing assets will break installs.


Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

Comment @coderabbitai help to get the list of available commands and usage tips.

Copy link
Collaborator

@RobotSail RobotSail left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@Maxusmusti Maxusmusti merged commit 28bf52a into main Oct 13, 2025
4 checks passed
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.

3 participants