File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -156,11 +156,15 @@ elif [[ $FRAMEWORK == "dynamo-vllm" ]]; then
156156 export MODEL_PATH=" /mnt/lustre01/models/kimi-k2.5-nvfp4"
157157 export SRT_SLURM_MODEL_PREFIX=" kimi-k2.5-nvfp4"
158158 elif [[ $MODEL_PREFIX == " dsv4" && $PRECISION == " fp4" ]]; then
159- # The FP4 checkpoint is staged on compute-visible Lustre. The former
160- # /mnt/numa1 path is no longer present on watchtower compute nodes;
161- # the lowercase Lustre sibling is the FP8 checkpoint, so keep the
162- # NVFP4 path explicit here.
163- export MODEL_PATH=" /mnt/lustre01/models/DeepSeek-V4-Pro-NVFP4/"
159+ if [[ " ${IS_AGENTIC:- 0} " == " 1" ]]; then
160+ # AgentX was tuned against the canonical DeepSeek-V4-Pro
161+ # checkpoint, which is staged on compute-visible Lustre.
162+ export MODEL_PATH=" /mnt/lustre01/models/deepseek-v4-pro"
163+ else
164+ # Existing fixed-sequence GB200 recipes use the NVIDIA ModelOpt
165+ # NVFP4 checkpoint.
166+ export MODEL_PATH=" /mnt/lustre01/models/DeepSeek-V4-Pro-NVFP4/"
167+ fi
164168 export SRT_SLURM_MODEL_PREFIX=" deepseek-v4-pro"
165169 elif [[ $MODEL_PREFIX == " minimaxm2.5" && $PRECISION == " fp4" ]]; then
166170 export MODEL_PATH=" /mnt/lustre01/models/MiniMax-M2.5-NVFP4"
You can’t perform that action at this time.
0 commit comments