make sure ninja is installed

This commit is contained in:
Wing Lian
2025-03-21 06:57:06 -04:00
parent 2518a9b2a2
commit e7532c9b0c

View File

@@ -31,9 +31,9 @@ RUN if [ "$NIGHTLY_BUILD" = "true" ] ; then \
sed -i 's#^datasets.*#datasets @ git+https://github.com/huggingface/datasets.git@main#' requirements.txt; \
fi
RUN pip3 install -U packaging setuptools wheel
RUN pip3 install -U packaging setuptools wheel ninja
RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \
pip install deepspeed; \
pip install --no-build-isolation deepspeed; \
pip install --no-build-isolation -e .[flash-attn,optimizers,ray,$AXOLOTL_EXTRAS] $AXOLOTL_ARGS; \
else \
pip install deepspeed; \