Skip to content

Commit 5e874ad

Browse files
mgoinzhewenl
authored andcommitted
Use float32 for test_completion.py (vllm-project#22385)
Signed-off-by: Michael Goin <[email protected]>
1 parent 34575d1 commit 5e874ad

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/v1/entrypoints/openai/test_completion.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,8 @@
2020
@pytest.fixture(scope="module")
2121
def default_server_args():
2222
return [
23-
# use half precision for speed and memory savings in CI environment
2423
"--dtype",
25-
"bfloat16",
24+
"float32",
2625
"--max-model-len",
2726
"2048",
2827
"--max-num-seqs",

0 commit comments

Comments
 (0)