Update args

vllm-project · Jul 17, 2024 · bef87a2 · bef87a2
1 parent e22aea4
commit bef87a2
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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