From 64ee6660fb582418d96a3fa2c37d428197c98937 Mon Sep 17 00:00:00 2001 From: ByronHsu Date: Fri, 6 Dec 2024 02:07:59 +0000 Subject: [PATCH] base gpu id --- rust/py_test/test_launch_server.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/rust/py_test/test_launch_server.py b/rust/py_test/test_launch_server.py index 13fcc8f4e4c..abcf4cb59e1 100644 --- a/rust/py_test/test_launch_server.py +++ b/rust/py_test/test_launch_server.py @@ -90,6 +90,8 @@ def popen_launch_server( host, "--port", port, + "--base-gpu-id", + "1", ] process = subprocess.Popen(command, stdout=None, stderr=None)