@@ -12,6 +12,8 @@ source ~/.bashrc
1212micromamba create -n slime python=3.12 pip -c conda-forge -y
1313micromamba activate slime
1414export CUDA_HOME=" $CONDA_PREFIX "
15+ export SGLANG_COMMIT=" 24c91001cf99ba642be791e099d358f4dfe955f5"
16+ export MEGATRON_COMMIT=" 3714d81d418c9f1bca4594fc35f9e8289f652862"
1517
1618export BASE_DIR=${BASE_DIR:- " /root" }
1719cd $BASE_DIR
@@ -27,7 +29,7 @@ pip install torch==2.9.1 torchvision==0.24.1 torchaudio==2.9.1 --index-url https
2729# install sglang
2830git clone https://github.com/sgl-project/sglang.git
2931cd sglang
30- git checkout 5e2cda6158e670e64b926a9985d65826c537ac82
32+ git checkout ${SGLANG_COMMIT}
3133# Install the python packages
3234pip install -e " python[all]"
3335
@@ -53,7 +55,7 @@ pip install nvidia-modelopt[torch]>=0.37.0 --no-build-isolation
5355# megatron
5456cd $BASE_DIR
5557git clone https://github.com/NVIDIA/Megatron-LM.git --recursive && \
56- cd Megatron-LM/ && git checkout core_v0.14.0 && \
58+ cd Megatron-LM/ && git checkout ${MEGATRON_COMMIT} && \
5759 pip install -e .
5860
5961# https://github.com/pytorch/pytorch/issues/168167
7577
7678# apply patch
7779cd $BASE_DIR /sglang
78- git apply $SLIME_DIR /docker/patch/v0.5.6 /sglang.patch
80+ git apply $SLIME_DIR /docker/patch/v0.5.7 /sglang.patch
7981cd $BASE_DIR /Megatron-LM
80- git apply $SLIME_DIR /docker/patch/v0.5.6 /megatron.patch
82+ git apply $SLIME_DIR /docker/patch/v0.5.7 /megatron.patch
0 commit comments