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.

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

rebased Flex attention support #2363

Open
wants to merge 8 commits into
base: main
Choose a base branch
from
Open

rebased Flex attention support #2363

wants to merge 8 commits into from

Conversation

bursteratom
Copy link
Collaborator

Description

Awaiting upstream: huggingface/transformers#36103

Motivation and Context

How has this been tested?

Screenshots (if appropriate)

Types of changes

Social Handles (Optional)

@bursteratom
Copy link
Collaborator Author

huggingface/transformers#36643 got merged!

@bursteratom bursteratom requested a review from winglian March 11, 2025 13:04
@winglian
Copy link
Collaborator

Discussed with Sunny, hold on this until transformers cuts a new release with the upstreamed fix. In the meantime, this PR should add some smoke tests for e2e and point to transformers@main to verify the fixes.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants