Skip to content

fix: bump lm_eval to 0.4.11 for transformers 4.56+ compatibility - #225

Merged
mwiebe merged 1 commit into
aws-deadline:mainlinefrom
anika-suman-amazon:fix/lm-eval-version-bump
May 22, 2026
Merged

mwiebe merged 1 commit into
aws-deadline:mainlinefrom
anika-suman-amazon:fix/lm-eval-version-bump

Conversation

@anika-suman-amazon

Copy link
Copy Markdown
Contributor

What was the problem/requirement? (What/Why)

The conda-forge vllm=0.19.1 package was rebuilt with a dependency on transformers>=4.56.0. In transformers 4.56+, AutoModelForVision2Seq was removed/renamed. lm_eval=0.4.9 still references this attribute at import time, causing an immediate crash before any evaluation runs.

What was the solution? (How)

Bumped lm_eval from 0.4.9 to 0.4.11 in the CondaPackages parameter default in template.yaml. Version 0.4.11 is compatible with transformers>=4.56.0.

What is the impact of this change?

Without this fix, the job bundle fails during benchmark evaluation due to the import error. With this fix, the bundle works as intended again.

How was this change tested?

Ran the job bundle -- all 3 default models (Qwen/Qwen2.5-0.5B, Qwen/Qwen2.5-1.5B, EleutherAI/pythia-1.4b) evaluated successfully across all 4 default benchmarks (hellaswag, arc_easy, arc_challenge, winogrande). The Aggregate step produced the expected leaderboard.csv and leaderboard.md with results matching the README examples.

Was this change documented?

No documentation changes needed, the README does not reference the specific lm_eval version.


By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@anika-suman-amazon
anika-suman-amazon requested a review from a team May 22, 2026 17:55
@github-actions github-actions Bot added the waiting-on-maintainers Waiting on the maintainers to review. label May 22, 2026
Signed-off-by: Anika Suman <anika-suman-amazon@users.noreply.github.com>
@mwiebe
mwiebe force-pushed the fix/lm-eval-version-bump branch from 299932a to 330a952 Compare May 22, 2026 23:24
@mwiebe
mwiebe merged commit b215baf into aws-deadline:mainline May 22, 2026
4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

waiting-on-maintainers Waiting on the maintainers to review.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants