diff --git a/build_conda.sh b/build_conda.sh index 43564311a..ac19b960d 100644 --- a/build_conda.sh +++ b/build_conda.sh @@ -50,7 +50,7 @@ NVCC_APPEND_FLAGS="--threads 4" \ pip install git+https://github.com/fzyzcjy/torch_memory_saver.git@dc6876905830430b5054325fa4211ff302169c6b --no-cache-dir --force-reinstall pip install git+https://github.com/fzyzcjy/Megatron-Bridge.git@dev_rl --no-build-isolation -pip install nvidia-modelopt[torch]>=0.37.0 --no-build-isolation +pip install "nvidia-modelopt[torch]>=0.37.0" --no-build-isolation # megatron cd $BASE_DIR @@ -80,4 +80,4 @@ pip install "numpy<2" cd $BASE_DIR/sglang git apply $SLIME_DIR/docker/patch/v0.5.7/sglang.patch cd $BASE_DIR/Megatron-LM -git apply $SLIME_DIR/docker/patch/v0.5.7/megatron.patch \ No newline at end of file +git apply $SLIME_DIR/docker/patch/v0.5.7/megatron.patch