CI fixes

axolotl-ai-cloud · Feb 6, 2024 · 4609e3b · 4609e3b
1 parent 67c70d1
commit 4609e3b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -48,6 +48,7 @@ jobs:
 
       - name: Install dependencies
         run: |
+          pip3 install packaging
           pip3 install -U -e .
           pip3 install -r requirements-tests.txt
 

diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -18,6 +18,8 @@ RUN git clone --depth=1 https://github.com/OpenAccess-AI-Collective/axolotl.git
 
 WORKDIR /workspace/axolotl
 
+RUN 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]; \