lightning-thunder
lightning-thunder copied to clipboard
Fix benchmark job
What does this PR do?
Fixes the benchmark job, see https://github.com/Lightning-AI/lightning-thunder/pull/2607/files.
Required bumping transformers to run with torch==2.8.0+cu12.8, which fixes https://github.com/Lightning-AI/lightning-thunder/issues/2302 (although still not supporting vmap)
PR review
Anyone in the community is free to review the PR once the tests have passed. If we didn't discuss your PR in Github issues there's a high chance it will not be merged.
Did you have fun?
Make sure you had fun coding 🙃