From 5a5d7f1cb1fb3c05e074074b293c4db0aa687024 Mon Sep 17 00:00:00 2001 From: Chirag Jain Date: Thu, 2 May 2024 16:09:38 +0000 Subject: [PATCH] Bump axolotl version - removes sigint handler --- Dockerfile | 2 +- Dockerfile-notebook | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 12e98ca..5768865 100644 --- a/Dockerfile +++ b/Dockerfile @@ -9,7 +9,7 @@ RUN mkdir -p /packages && \ cd /packages && \ git clone https://github.com/truefoundry/axolotl && \ cd axolotl/ && \ - git checkout 7ac62f5fa6b3df526a7d0fed7c711faa20df12b0 + git checkout 4e8264e937571c53b9dc75345a14d4b9b9d68c4f RUN cd /packages/axolotl/ && \ MAX_JOBS=1 NVCC_APPEND_FLAGS="--threads 1" pip install -U --no-build-isolation -e .[flash-attn,mamba-ssm,fused-dense-lib] && \ pip install --no-cache-dir -U -r /tmp/requirements.txt && \ diff --git a/Dockerfile-notebook b/Dockerfile-notebook index 008c792..a5c18db 100644 --- a/Dockerfile-notebook +++ b/Dockerfile-notebook @@ -21,7 +21,7 @@ USER jovyan RUN cd /packages && \ git clone https://github.com/truefoundry/axolotl && \ cd axolotl/ && \ - git checkout 7ac62f5fa6b3df526a7d0fed7c711faa20df12b0 + git checkout 4e8264e937571c53b9dc75345a14d4b9b9d68c4f RUN cd /packages/axolotl/ && \ MAX_JOBS=1 NVCC_APPEND_FLAGS="--threads 1" pip install -U --no-build-isolation -e .[flash-attn,mamba-ssm,fused-dense-lib] && \ pip install --no-cache-dir -U -r /tmp/llm-finetune/notebook-requirements.txt