tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

CI fixes

Browse Source
This commit is contained in:
Maxime
2024-02-06 16:38:00 +01:00
parent 67c70d1954
commit 4609e3b166
2 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.github/workflows/tests.yml vendored
View File

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

2
docker/Dockerfile
View File

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