87 lines
2.9 KiB
Python
87 lines
2.9 KiB
Python
"""Launch the inference server."""
|
|
|
|
import asyncio
|
|
import os
|
|
import sys
|
|
import warnings
|
|
|
|
from sglang.srt.arg_groups.overrides import resolving_view
|
|
from sglang.srt.plugins import load_plugins
|
|
from sglang.srt.server_args import prepare_server_args
|
|
from sglang.srt.utils import kill_process_tree
|
|
from sglang.srt.utils.common import suppress_noisy_warnings
|
|
|
|
suppress_noisy_warnings()
|
|
|
|
|
|
def run_server(server_args):
|
|
"""Run the server based on the gRPC flags and server_args.encoder_only."""
|
|
# smg-grpc-servicer reads the raw `server_args.tokenizer_path`, which
|
|
# resolution defaults into the stash but never writes back. Seed it here,
|
|
# before resolve_once() seals the record.
|
|
if (
|
|
server_args.smg_grpc_mode or server_args.grpc_mode
|
|
) and server_args.tokenizer_path is None:
|
|
server_args.tokenizer_path = server_args.model_path
|
|
|
|
# The flags dispatched on below are decided by resolution (`--grpc-mode`
|
|
# folds into `smg_grpc_mode`), and `prepare_server_args` returns raw input.
|
|
server_args.resolve_once()
|
|
cfg = resolving_view(server_args)
|
|
|
|
if cfg.encoder_only:
|
|
# For encoder disaggregation
|
|
if cfg.smg_grpc_mode or cfg.grpc_mode:
|
|
from sglang.srt.disaggregation.encoder.grpc_server import (
|
|
serve_grpc_encoder,
|
|
)
|
|
|
|
asyncio.run(serve_grpc_encoder(server_args))
|
|
else:
|
|
from sglang.srt.disaggregation.encoder.http_server import launch_server
|
|
|
|
launch_server(server_args)
|
|
elif cfg.smg_grpc_mode:
|
|
# Legacy SMG gRPC server (--smg-grpc-mode, or the deprecated --grpc-mode
|
|
# which __post_init__ folds into smg_grpc_mode). The native Rust gRPC
|
|
# server is a separate path, enabled by --grpc-port, that starts
|
|
# alongside the default HTTP server below.
|
|
from sglang.srt.entrypoints.grpc_server import serve_grpc
|
|
|
|
asyncio.run(serve_grpc(server_args))
|
|
elif cfg.use_ray:
|
|
# Ray mode: HTTP mode with Ray backend.
|
|
try:
|
|
from sglang.srt.ray.http_server import launch_server
|
|
except ImportError:
|
|
raise ImportError(
|
|
"Ray is required for --use-ray mode. "
|
|
"Install it with: pip install 'sglang[ray]'"
|
|
)
|
|
|
|
launch_server(server_args)
|
|
else:
|
|
# Default mode: HTTP mode.
|
|
from sglang.srt.entrypoints.http_server import launch_server
|
|
|
|
launch_server(server_args)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
warnings.warn(
|
|
"'python -m sglang.launch_server' is still supported, but "
|
|
"'sglang serve' is the recommended entrypoint.\n"
|
|
" Example: sglang serve --model-path <model> [options]",
|
|
UserWarning,
|
|
stacklevel=1,
|
|
)
|
|
|
|
load_plugins()
|
|
|
|
server_args = prepare_server_args(sys.argv[1:])
|
|
|
|
try:
|
|
run_server(server_args)
|
|
finally:
|
|
kill_process_tree(os.getpid(), include_parent=False)
|