Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ base:
name: dsr1-fp4-b200-sglang-mtp-8k1k
model:
path: hf:nvidia/DeepSeek-R1-0528-FP4-V2
container: lmsysorg/sglang:v0.5.16-cu130
container: lmsysorg/sglang:v0.5.21-cu130@sha256:b1259f3ea3275f66237c498ea388919729018bc9f01c3d638391e06e2cf3f469
precision: fp4
resources:
gpu_type: b200
Expand All @@ -25,7 +25,7 @@ base:
trust-remote-code: true
tensor-parallel-size: 4
data-parallel-size: 1
cuda-graph-max-bs: 256
cuda-graph-max-bs-decode: 256
max-running-requests: 256
mem-fraction-static: 0.85
kv-cache-dtype: fp8_e4m3
Expand All @@ -34,7 +34,7 @@ base:
quantization: modelopt_fp4
enable-flashinfer-allreduce-fusion: true
scheduler-recv-interval: 10
disable-piecewise-cuda-graph: true
cuda-graph-backend-prefill: disabled
attention-backend: trtllm_mla
moe-runner-backend: flashinfer_trtllm
stream-interval: 10
Expand Down
2 changes: 1 addition & 1 deletion inferencex-e2e/configs/nvidia-master.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -893,7 +893,7 @@ dsr1-fp4-b200-sglang:
# - { tp: 8, ep: 8, offloading: none, conc-list: [1, 2, 4, 8, 12, 16, 32, 64, 128, 256, 512] }

dsr1-fp4-b200-sglang-mtp:
image: lmsysorg/sglang:v0.5.16-cu130
image: lmsysorg/sglang:v0.5.21-cu130@sha256:b1259f3ea3275f66237c498ea388919729018bc9f01c3d638391e06e2cf3f469
model: nvidia/DeepSeek-R1-0528-FP4-V2
model-prefix: dsr1
runner: cluster:b200-nscale
Expand Down
Loading