Skip to content

Commit 08afed5

Browse files
authored
[cb][tests] 🐛 fix bug in test utils, please merge ASAP (#367)
### [cb][tests] 🐛 fix bug in test utils, please merge ASAP during some test script refactoring, the parameter `max_model_len` was overridden to always be 256, even if set differently. Signed-off-by: Yannick Schnider <[email protected]>
1 parent 1037b62 commit 08afed5

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

tests/scheduling_utils.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,8 +60,6 @@ def check_scheduler_inference_steps(
6060
if use_cb:
6161
monkeypatch.setenv("VLLM_SPYRE_USE_CB", "1")
6262

63-
max_model_len = 256
64-
6563
# Input parameters sanity check, not actual testing
6664
# ------
6765
if not (len(prompts_lengths) == len(seqs_max_tokens)

0 commit comments

Comments
 (0)