Make JAIS compatible with Transformers v5 (#34264)

Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
2026-02-11 13:30:37 +01:00
parent 1e9204bff3
commit 0f5e55e7a8
1 changed files with 0 additions and 1 deletions
--- a/vllm/model_executor/models/jais.py
+++ b/vllm/model_executor/models/jais.py
@@ -244,7 +244,6 @@ class JAISModel(nn.Module):
        quant_config = vllm_config.quant_config

        self.config = config
-        assert not config.add_cross_attention
        assert not config.scale_attn_by_inverse_layer_idx
        assert not config.reorder_and_upcast_attn
        self.embed_dim = config.hidden_size