From bef87a2c6e86aab5ce8db0cb13b48260c88ff040 Mon Sep 17 00:00:00 2001 From: DarkLight1337 Date: Wed, 17 Jul 2024 07:46:28 +0000 Subject: [PATCH] Update args --- tests/entrypoints/openai/test_basic.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) 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