Skip to content

Make JAIS compatible with Transformers v5#34264

Merged
DarkLight1337 merged 2 commits intovllm-project:mainfrom
hmellor:fix-jais-v5
Feb 11, 2026
Merged

Make JAIS compatible with Transformers v5#34264
DarkLight1337 merged 2 commits intovllm-project:mainfrom
hmellor:fix-jais-v5

Conversation

@hmellor
Copy link
Member

@hmellor hmellor commented Feb 10, 2026

add_cross_attention was an attribute of PreTrainedConfig that got removed in huggingface/transformers#41541

This attribute was never explicitly in the JAIS config and the checkpoints we use for testing in CI do not set it.

Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Copy link
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

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

Code Review

This pull request correctly updates the JAIS model to be compatible with Transformers v5 by removing the assert not config.add_cross_attention line. This assertion was for an attribute that no longer exists in the PreTrainedConfig in the newer Transformers version, making its removal a necessary and appropriate change for compatibility.

@DarkLight1337 DarkLight1337 enabled auto-merge (squash) February 11, 2026 10:29
@github-actions github-actions bot added the ready ONLY add when PR is ready to merge/full CI is needed label Feb 11, 2026
@DarkLight1337 DarkLight1337 merged commit 0f5e55e into vllm-project:main Feb 11, 2026
53 checks passed
@hmellor hmellor deleted the fix-jais-v5 branch February 11, 2026 14:12
eldarkurtic pushed a commit to eldarkurtic/vllm that referenced this pull request Feb 19, 2026
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Signed-off-by: Eldar Kurtic <research@neuralmagic.com>
llsj14 pushed a commit to llsj14/vllm that referenced this pull request Mar 1, 2026
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
tunglinwood pushed a commit to tunglinwood/vllm that referenced this pull request Mar 4, 2026
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ready ONLY add when PR is ready to merge/full CI is needed

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants