fix dockerfile

This commit is contained in:
tastelikefeet
2025-08-13 15:22:02 +08:00
parent 5e76b351a3
commit 743304f3da

View File

@@ -76,7 +76,7 @@ RUN sh /tmp/install.sh {version_args} && \
RUN if [ "$INSTALL_MS_DEPS" = "True" ]; then \
pip install --no-cache-dir huggingface-hub transformers -U; \
pip install --no-cache-dir huggingface-hub transformers peft -U; \
fi; \
if [ "$INSTALL_MEGATRON_DEPS" = "True" ]; then \
pip install liger_kernel nvitop pre-commit transformers huggingface-hub -U && \