This commit is contained in:
Jintao Huang
2025-10-13 14:46:52 +08:00
parent b5c33c2e53
commit 5a7e52c2f4

View File

@@ -87,7 +87,7 @@ RUN if [ "$IMAGE_TYPE" = "swift" ]; then \
pip install git+https://github.com/NVIDIA/Megatron-LM.git@core_r0.13.0; \ pip install git+https://github.com/NVIDIA/Megatron-LM.git@core_r0.13.0; \
elif [ "$IMAGE_TYPE" = "gpu" ]; then \ elif [ "$IMAGE_TYPE" = "gpu" ]; then \
pip install "transformers<4.56" "trl<0.23" --no-dependencies; \ pip install "transformers<4.56" "trl<0.23" --no-dependencies; \
else else \
pip install --no-cache-dir huggingface-hub transformers peft -U; \ pip install --no-cache-dir huggingface-hub transformers peft -U; \
fi fi