Merge branch 'mps' of https://github.com/maximegmd/axolotl into mps

axolotl-ai-cloud · Feb 7, 2024 · cd829e8 · cd829e8
2 parents 73e72db + 980e7aa
commit cd829e8
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -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

diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -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 \

diff --git a/docker/Dockerfile-tests b/docker/Dockerfile-tests
@@ -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]; \