diff --git a/tests/entrypoints/openai/test_basic.py b/tests/entrypoints/openai/test_basic.py index de329d8cc6659..2c721d9ba7609 100644 --- a/tests/entrypoints/openai/test_basic.py +++ b/tests/entrypoints/openai/test_basic.py @@ -14,8 +14,6 @@ @pytest.fixture(scope="module") def server(): args = [ - "--model", - MODEL_NAME, # use half precision for speed and memory savings in CI environment "--dtype", "bfloat16", @@ -26,7 +24,7 @@ def server(): "128", ] - with RemoteOpenAIServer(args) as remote_server: + with RemoteOpenAIServer(MODEL_NAME, args) as remote_server: yield remote_server