Skip to content

Commit bf52428

Browse files
author
chibu
committed
use cli bin
1 parent 36cba31 commit bf52428

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/automation/vllm/server.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,8 @@ def start_vllm_server(
2626
parsed_target = urlparse(target)
2727

2828
server_command = [
29-
f"{vllm_path}", "serve",
29+
#f"{vllm_path}", "serve",
30+
"vllm", "serve",
3031
model_id,
3132
"--host", parsed_target.hostname,
3233
"--port", str(parsed_target.port),
@@ -35,6 +36,7 @@ def start_vllm_server(
3536

3637
subprocess_env = os.environ.copy()
3738

39+
"""
3840
for k, v in vllm_args.items():
3941
if k.startswith("VLLM_"):
4042
subprocess_env[k] = str(v)
@@ -43,6 +45,7 @@ def start_vllm_server(
4345
server_command.append(f"--{k}")
4446
else:
4547
server_command.extend([f"--{k}", str(v)])
48+
"""
4649

4750

4851
server_log_file_name = f"{SERVER_LOG_PREFIX}_{task.id}.txt"

0 commit comments

Comments
 (0)