From 3e38ce58ef0356d44d1b1163e2b8a5ec1fb9cb5e Mon Sep 17 00:00:00 2001 From: Chirag Jain Date: Fri, 28 Jun 2024 19:18:40 +0000 Subject: [PATCH] update axolotl commit --- Dockerfile | 2 +- Dockerfile-notebook | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 548305a..6b50f39 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 6a3ca76d3876ba7d9de9480cd203a8356e7279a7 + git checkout d070b883055766197c5469a5cc51a3d91296579e 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 dc35a43..f9d88b8 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 6a3ca76d3876ba7d9de9480cd203a8356e7279a7 + git checkout d070b883055766197c5469a5cc51a3d91296579e 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