Skip to content

Commit 6c9a87c

Browse files
authored
pin accelerate so it works with llama2 (axolotl-ai-cloud#330)
1 parent 894cba0 commit 6c9a87c

File tree

2 files changed

+1
-5
lines changed

2 files changed

+1
-5
lines changed

docker/Dockerfile

-4
Original file line numberDiff line numberDiff line change
@@ -11,10 +11,6 @@ RUN apt-get update && \
1111

1212
WORKDIR /workspace
1313

14-
RUN pip3 install --force-reinstall "peft @ git+https://github.com/huggingface/peft.git@main" \
15-
"accelerate @ git+https://github.com/huggingface/accelerate.git@main" \
16-
"transformers @ git+https://github.com/huggingface/transformers.git@main"
17-
1814
RUN git clone --depth=1 https://github.com/OpenAccess-AI-Collective/axolotl.git
1915
# If AXOLOTL_EXTRAS is set, append it in brackets
2016
RUN cd axolotl && \

requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
peft @ git+https://github.com/huggingface/peft.git
22
transformers @ git+https://github.com/huggingface/transformers.git
33
bitsandbytes>=0.39.0
4-
accelerate==0.21.0
4+
accelerate @ git+https://github.com/huggingface/accelerate@2a289f6108e77a77a4efffb3f6316bc98538413b
55
addict
66
fire
77
PyYAML==6.0

0 commit comments

Comments
 (0)