mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-16 16:27:45 +01:00
fix
This commit is contained in:
@@ -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
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user