LLaDA2.0-mini on one GPU
This recipe uses one NVIDIA GPU with compute capability 9.0 or newer and enough memory for the model and KV cache. Complete the Docker installation first.
Launch configuration
Section titled “Launch configuration”CUDA_VISIBLE_DEVICES=0 fluxserve serve \ --model inclusionAI/LLaDA2.0-mini \ --host 127.0.0.1 --port 8000 \ --tp-size 1 --dp-size 1 --ep-size 1 \ --max-num-seqs 4 \ --max-model-len 4096 \ --block-length 64 \ --threshold 0.95 \ --parallel-decoding threshold \ --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 4Configuration notes
Section titled “Configuration notes”--max-num-seqslimits concurrent sequences;--max-model-lensets the context-length limit.--block-lengthand--thresholdconfigure block diffusion decoding.- The listed CUDA graph capture sizes cover this recipe’s batch sizes.
- If model loading or cache allocation exceeds available memory, use a suitable GPU or reduce context length and concurrency before benchmarking.
Verify and benchmark
Section titled “Verify and benchmark”Use the readiness check and chat request in the quickstart, then follow the online benchmark.
Source revision: 7183e38.
