Skip to content

Commit 9e329c2

Browse files
committed
fix: canonical ckpt for agentic
1 parent 6910a9f commit 9e329c2

1 file changed

Lines changed: 9 additions & 5 deletions

File tree

runners/launch_gb200-nv.sh

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff 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"

0 commit comments

Comments
 (0)