Fixing flex attention for torch=2.6.0 #37285

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

Merged

ArthurZucker merged 6 commits into huggingface:main from SalmanMohammadi:flex_2_6_fix

Apr 7, 2025

Merged

Fixing flex attention for torch=2.6.0 #37285

Merge branch 'main' into flex_2_6_fix

CircleCI Checks / setup_and_quality succeeded Apr 7, 2025 in 4m 15s

Workflow: setup_and_quality

View CircleCI Workflow

View more details on CircleCI Checks