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

@@ -48,6 +48,7 @@ jobs:
- name: Install dependencies
run: |
pip3 install --upgrade pip
pip3 install packaging
pip3 install -U -e .
pip3 install -r requirements-tests.txt

View File

@@ -18,7 +18,8 @@ RUN git clone --depth=1 https://github.com/OpenAccess-AI-Collective/axolotl.git
WORKDIR /workspace/axolotl
RUN pip install packaging
RUN pip install --upgrade pip && \
pip install packaging
# If AXOLOTL_EXTRAS is set, append it in brackets
RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \

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]; \