install packaging for various tests

This commit is contained in:
Wing Lian
2024-02-07 09:43:31 -05:00
parent 4609e3b166
commit 980e7aa44d
3 changed files with 6 additions and 1 deletions

View File

@@ -22,6 +22,9 @@ WORKDIR /workspace/axolotl
RUN git fetch origin +$GITHUB_REF && \
git checkout FETCH_HEAD
RUN pip install --upgrade pip && \
pip install packaging \
# If AXOLOTL_EXTRAS is set, append it in brackets
RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \
pip install -e .[deepspeed,flash-attn,mamba-ssm,$AXOLOTL_EXTRAS]; \