Skip to content

Commit 6c380d3

Browse files
committed
Enable up to 32k
Signed-off-by: Rafael Vasquez <[email protected]>
1 parent 4ff087e commit 6c380d3

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/e2e/test_spyre_cb.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def test_long_context_batches(
123123
monkeypatch.setenv("VLLM_SPYRE_DYNAMO_BACKEND", backend)
124124
monkeypatch.setenv("VLLM_SPYRE_OVERRIDE_SIGNALS_HANDLER", "1")
125125

126-
max_model_len = 16384
126+
max_model_len = 32768
127127
max_num_seqs = 32
128128
max_tokens = 10
129129

@@ -134,6 +134,7 @@ def test_long_context_batches(
134134
(8, 3000),
135135
(4, 5000),
136136
(2, 9000),
137+
(1, 17000),
137138
]
138139

139140
vllm_model = LLM(model=model.name,

0 commit comments

Comments
 (0)