54 lines
1.5 KiB
YAML
54 lines
1.5 KiB
YAML
services:
|
|
sglang:
|
|
image: ymkymx/sglang:dsv41-pd-visioncp-db7d2cb7d-fix
|
|
container_name: ds41-pd-prefill
|
|
shm_size: "32gb"
|
|
ipc: host
|
|
pid: host
|
|
privileged: true
|
|
network_mode: host
|
|
|
|
environment:
|
|
- CUDA_VISIBLE_DEVICES=0,1,2,3
|
|
- SGLANG_RAGGED_VERIFY_MODE=static
|
|
- MC_INTRANODE_NVLINK=true
|
|
- MC_INTRA_NVLINK=true
|
|
- SGLANG_MOONCAKE_SEND_AUX_TCP=1
|
|
volumes:
|
|
- /data:/data
|
|
- /data/ymk/cache/sglang:/root/.cache/sglang
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: all
|
|
capabilities: [gpu]
|
|
command: >-
|
|
sglang serve
|
|
--trust-remote-code
|
|
--model-path /data/models/DeepSeek-V4.1-Flash
|
|
--tp 4
|
|
--ep-size 4
|
|
--mem-fraction-static 0.75
|
|
--attention-backend dsv4
|
|
--moe-runner-backend flashinfer_mxfp4
|
|
--cuda-graph-max-bs-decode 64
|
|
--reasoning-parser auto
|
|
--tool-call-parser auto
|
|
--default-chat-template-kwargs '{"thinking": true, "reasoning_effort": "high"}'
|
|
--tokenizer-worker-num 8
|
|
--host 0.0.0.0
|
|
--port 30000
|
|
--enable-cache-report
|
|
--enable-metrics
|
|
--speculative-algorithm DSPARK
|
|
--speculative-dspark-block-size 5
|
|
--enable-hierarchical-cache
|
|
--hicache-ratio 2.5
|
|
--hicache-write-policy write_back
|
|
--enable-prefill-cp
|
|
--cp-strategy interleave
|
|
--disaggregation-mode prefill
|
|
--disaggregation-transfer-backend mooncake
|