Skip to content

Commit e138bdf

Browse files
reviewer commemnts
Signed-off-by: morrison-turnansky <mturnans@redhat.com>
1 parent 79708c1 commit e138bdf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/compile/test_config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ def test_stock_torch_compile(vllm_runner, monkeypatch):
9999
# loading the model causes compilation (if enabled) to happen
100100
vllm_runner(
101101
"facebook/opt-125m",
102-
compilation_config={"mode": 1},
102+
compilation_config={"mode": CompilationMode.STOCK_TORCH_COMPILE},
103103
gpu_memory_utilization=0.4,
104104
) as _,
105105
):

0 commit comments

Comments
 (0)