Server Arguments#
Common launch commands#
To enable multi-GPU tensor parallelism, add
--tp 2
. If it reports the error “peer access is not supported between these two devices”, add--enable-p2p-check
to the server launch command.
python -m sglang.launch_server --model-path meta-llama/Meta-Llama-3-8B-Instruct --tp 2
To enable multi-GPU data parallelism, add
--dp 2
. Data parallelism is better for throughput if there is enough memory. It can also be used together with tensor parallelism. The following command uses 4 GPUs in total. We recommend SGLang Router for data parallelism.
python -m sglang_router.launch_server --model-path meta-llama/Meta-Llama-3-8B-Instruct --dp 2 --tp 2
If you see out-of-memory errors during serving, try to reduce the memory usage of the KV cache pool by setting a smaller value of
--mem-fraction-static
. The default value is0.9
.
python -m sglang.launch_server --model-path meta-llama/Meta-Llama-3-8B-Instruct --mem-fraction-static 0.7
See hyperparameter tuning on tuning hyperparameters for better performance.
For docker and Kubernetes runs, you need to set up shared memory which is used for communication between processes. See
--shm-size
for docker and/dev/shm
size update for Kubernetes manifests.If you see out-of-memory errors during prefill for long prompts, try to set a smaller chunked prefill size.
python -m sglang.launch_server --model-path meta-llama/Meta-Llama-3-8B-Instruct --chunked-prefill-size 4096
To enable
torch.compile
acceleration, add--enable-torch-compile
. It accelerates small models on small batch sizes. By default, the cache path is located at/tmp/torchinductor_root
, you can customize it using environment variableTORCHINDUCTOR_CACHE_DIR
. For more details, please refer to PyTorch official documentation and Enabling cache for torch.compile.To enable torchao quantization, add
--torchao-config int4wo-128
. It supports other quantization strategies (INT8/FP8) as well.To enable fp8 weight quantization, add
--quantization fp8
on a fp16 checkpoint or directly load a fp8 checkpoint without specifying any arguments.To enable fp8 kv cache quantization, add
--kv-cache-dtype fp8_e5m2
.If the model does not have a chat template in the Hugging Face tokenizer, you can specify a custom chat template.
To run tensor parallelism on multiple nodes, add
--nnodes 2
. If you have two nodes with two GPUs on each node and want to run TP=4, letsgl-dev-0
be the hostname of the first node and50000
be an available port, you can use the following commands. If you meet deadlock, please try to add--disable-cuda-graph
# Node 0
python -m sglang.launch_server --model-path meta-llama/Meta-Llama-3-8B-Instruct --tp 4 --dist-init-addr sgl-dev-0:50000 --nnodes 2 --node-rank 0
# Node 1
python -m sglang.launch_server --model-path meta-llama/Meta-Llama-3-8B-Instruct --tp 4 --dist-init-addr sgl-dev-0:50000 --nnodes 2 --node-rank 1
Please consult the documentation below to learn more about the parameters you may provide when launching a server.
Model, processor and tokenizer#
model_path
: Path to the model that will be served.tokenizer_path
: Defaults to themodel_path
.tokenizer_mode
: By defaultauto
, see here for different mode.load_format
: The format the weights are loaded in. Defaults to*.safetensors
/*.bin
.trust_remote_code
: IfTrue
, will use locally cached config files, otherwise use remote configs in HuggingFace.dtype
: Dtype used for the model, defaults tobfloat16
.kv_cache_dtype
: Dtype of the kv cache, defaults to thedtype
.context_length
: The number of tokens our model can process including the input. Note that extending the default might lead to strange behavior.device
: The device we put the model, defaults tocuda
.chat_template
: The chat template to use. Deviating from the default might lead to unexpected responses. For multi-modal chat templates, refer to here. Make sure the correctchat_template
is passed, or performance degradation may occur!!!!is_embedding
: Set to true to perform embedding / encode and reward tasks.revision
: Adjust if a specific version of the model should be used.skip_tokenizer_init
: Set to true to provide the tokens to the engine and get the output tokens directly, typically used in RLHF. Please see this example for reference.json_model_override_args
: Override model config with the provided JSON.delete_ckpt_after_loading
: Delete the model checkpoint after loading the model.disable_fast_image_processor
: Adopt base image processor instead of fast image processor(which is by default). For more detail, see: https://huggingface.co/docs/transformers/main/en/main_classes/image_processor#image-processor
Serving: HTTP & API#
HTTP Server configuration#
port
andhost
: Setup the host for HTTP server. By defaulthost: str = "127.0.0.1"
andport: int = 30000
API configuration#
api_key
: Sets an API key for the server and the OpenAI-compatible API.file_storage_path
: Directory for storing uploaded or generated files from API calls.enable_cache_report
: If set, includes detailed usage of cached tokens in the response usage.
Parallelism#
Tensor parallelism#
tp_size
: The number of GPUs the model weights get sharded over. Mainly for saving memory rather than for high throughput, see this blogpost.
Data parallelism#
dp_size
: Will be deprecated. The number of data-parallel copies of the model. SGLang router is recommended instead of the current naive data parallel.load_balance_method
: Will be deprecated. Load balancing strategy for data parallel requests.
Expert parallelism#
enable_ep_moe
: Enables expert parallelism that distributes the experts onto multiple GPUs for MoE models.ep_size
: The size of EP. Please shard the model weights withtp_size=ep_size
, for detailed benchmarking refer to this PR. If not set,ep_size
will be automatically set totp_size
.enable_deepep_moe
: Enables expert parallelism that distributes the experts onto multiple GPUs for DeepSeek-V3 model based on deepseek-ai/DeepEP.deepep_mode
: Select the mode when enable DeepEP MoE, could benormal
,low_latency
orauto
. Default isauto
, which meanslow_latency
for decode batch andnormal
for prefill batch.
Memory and scheduling#
mem_fraction_static
: Fraction of the free GPU memory used for static memory like model weights and KV cache. If building KV cache fails, it should be increased. If CUDA runs out of memory, it should be decreased.max_running_requests
: The maximum number of requests to run concurrently.max_total_tokens
: The maximum number of tokens that can be stored into the KV cache. Use mainly for debugging.chunked_prefill_size
: Perform the prefill in chunks of these size. Larger chunk size speeds up the prefill phase but increases the VRAM consumption. If CUDA runs out of memory, it should be decreased.max_prefill_tokens
: Token budget of how many tokens to accept in one prefill batch. The actual number is the max of this parameter and thecontext_length
.schedule_policy
: The scheduling policy to control the processing order of waiting prefill requests in a single engine.schedule_conservativeness
: Can be used to decrease/increase the conservativeness of the server when taking new requests. Highly conservative behavior leads to starvation, but low conservativeness leads to slowed-down performance.cpu_offload_gb
: Reserve this amount of RAM in GB for offloading of model parameters to the CPU.
Other runtime options#
stream_interval
: Interval (in tokens) for streaming responses. Smaller values lead to smoother streaming, and larger values lead to better throughput.random_seed
: Can be used to enforce more deterministic behavior.watchdog_timeout
: Adjusts the watchdog thread’s timeout before killing the server if batch generation takes too long.download_dir
: Use to override the default Hugging Face cache directory for model weights.base_gpu_id
: Use to adjust first GPU used to distribute the model across available GPUs.allow_auto_truncate
: Automatically truncate requests that exceed the maximum input length.
Logging#
log_level
: Global log verbosity.log_level_http
: Separate verbosity level for the HTTP server logs (if unset, defaults tolog_level
).log_requests
: Logs the inputs and outputs of all requests for debugging.log_requests_level
: Ranges from 0 to 2: level 0 only shows some basic metadata in requests, level 1 and 2 show request details (e.g., text, images), and level 1 limits output to 2048 characters (if unset, defaults to0
).show_time_cost
: Prints or logs detailed timing info for internal operations (helpful for performance tuning).enable_metrics
: Exports Prometheus-like metrics for request usage and performance.decode_log_interval
: How often (in tokens) to log decode progress.
Multi-node distributed serving#
dist_init_addr
: The TCP address used for initializing PyTorch’s distributed backend (e.g.192.168.0.2:25000
).nnodes
: Total number of nodes in the cluster. Refer to how to run the Llama 405B model.node_rank
: Rank (ID) of this node among thennodes
in the distributed setup.
LoRA#
lora_paths
: You may provide a list of adapters to your model as a list. Each batch element will get model response with the corresponding lora adapter applied. Currentlycuda_graph
andradix_attention
are not supported with this option so you need to disable them manually. We are still working on through these issues.max_loras_per_batch
: Maximum number of LoRAs in a running batch including base model.lora_backend
: The backend of running GEMM kernels for Lora modules, can be one oftriton
orflashinfer
. Defaults to betriton
.
Kernel backend#
attention_backend
: This argument specifies the backend for attention computation and KV cache management, which can befa3
,flashinfer
,triton
, ortorch_native
. When deploying DeepSeek models, use this argument to specify the MLA backend.sampling_backend
: The backend for sampling.
Constrained Decoding#
Speculative decoding#
speculative_draft_model_path
: The draft model path for speculative decoding.speculative_algorithm
: The algorithm for speculative decoding. Currently only Eagle is supported. Note that the radix cache, chunked prefill, and overlap scheduler are disabled when using eagle speculative decoding.speculative_num_steps
: How many draft passes we run before verifying.speculative_num_draft_tokens
: The number of tokens proposed in a draft.speculative_eagle_topk
: The number of top candidates we keep for verification at each step for Eagle.speculative_token_map
: Optional, the path to the high frequency token list of FR-Spec, used for accelerating Eagle.
Double Sparsity#
enable_double_sparsity
: Enables double sparsity which increases throughput.ds_channel_config_path
: The double sparsity config. For a guide on how to generate the config for your model see this repo.ds_heavy_channel_num
: Number of channel indices to keep for each layer.ds_heavy_token_num
: Number of tokens used for attention during decode. Skip sparse decoding ifmin_seq_len
in batch < this number.ds_heavy_channel_type
: The type of heavy channels. Eitherq
,k
orqk
.ds_sparse_decode_threshold
: Don’t apply sparse decoding ifmax_seq_len
in batch < this threshold.
Debug options#
Note: We recommend to stay with the defaults and only use these options for debugging for best possible performance.
disable_radix_cache
: Disable Radix backend for prefix caching.disable_cuda_graph
: Disable cuda graph for model forward. Use if encountering uncorrectable CUDA ECC errors.disable_cuda_graph_padding
: Disable cuda graph when padding is needed. In other case still use cuda graph.disable_outlines_disk_cache
: Disable disk cache for outlines grammar backend.disable_custom_all_reduce
: Disable usage of custom all reduce kernel.disable_mla
: Disable Multi-Head Latent Attention for Deepseek model.disable_overlap_schedule
: Disable the Overhead-Scheduler.enable_nan_detection
: Turning this on makes the sampler print a warning if the logits containNaN
.enable_p2p_check
: Turns off the default of allowing always p2p check when accessing GPU.triton_attention_reduce_in_fp32
: In triton kernels this will cast the intermediate attention result tofloat32
.
Optimization#
Note: Some of these options are still in experimental stage.
enable_mixed_chunk
: Enables mixing prefill and decode, see this discussion.enable_dp_attention
: Enable Data Parallelism Attention for Deepseek models.enable_torch_compile
: Torch compile the model. Note that compiling a model takes a long time but have a great performance boost. The compiled model can also be cached for future use.torch_compile_max_bs
: The maximum batch size when usingtorch_compile
.cuda_graph_max_bs
: Adjust the maximum batchsize when using cuda graph. By default this is chosen for you based on GPU specifics.cuda_graph_bs
: The batch sizes to capture byCudaGraphRunner
. By default this is done for you.torchao_config
: Experimental feature that optimizes the model with torchao. Possible choices are: int8dq, int8wo, int4wo-<group_size>, fp8wo, fp8dq-per_tensor, fp8dq-per_row.triton_attention_num_kv_splits
: Use to adjust the number of KV splits in triton kernels. Default is 8.enable_flashinfer_mla
: Use the attention backend with FlashInfer MLA wrapper for DeepSeek models. This argument will be deprecated in the next release. Please use--attention_backend flashinfer
instead to enable FlashfIner MLA.flashinfer_mla_disable_ragged
: Disable the use of the ragged prefill wrapper for the FlashInfer MLA attention backend. Only use it when FlashInfer is being used as the MLA backend.