Skip to content
FluxServe

LLaDA2.0-flash on four GPUs

Complete the Docker installation first. This recipe targets four NVIDIA H100 GPUs with sufficient memory for the checkpoint and cache.

TP=4, EP=4, Paged Flashinfer (H100, 4 x SM90)

Terminal window
CUDA_VISIBLE_DEVICES=0,1,2,3 fluxserve serve \
--model inclusionAI/LLaDA2.0-flash \
--host 127.0.0.1 \
--port 8000 \
--tp-size 4 \
--dp-size 1 \
--ep-size 4 \
--gpu-memory-utilization 0.8 \
--max-num-seqs 8 \
--max-model-len 16384 \
--block-length 64 \
--parallel-decoding threshold \
--threshold 0.95 \
--attention-backend flashinfer \
--kv-cache-layout paged \
--scheduler-policy paged \
--use-decode-cuda-graph \
--cuda-graph-decode-mode padded \
--cuda-graph-capture-bs 1 2 4 8

After model initialization, use the health check and request in the quickstart, replacing the request model with inclusionAI/LLaDA2.0-flash. See benchmarking to measure the running service.


Source revision: 7183e38.