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

View File

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

View File

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