winglian commited on
Commit
6c9a87c
1 Parent(s): 894cba0

pin accelerate so it works with llama2 (#330)

Browse files
Files changed (2) hide show
  1. docker/Dockerfile +0 -4
  2. requirements.txt +1 -1
docker/Dockerfile CHANGED
@@ -11,10 +11,6 @@ RUN apt-get update && \
11
 
12
  WORKDIR /workspace
13
 
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
-
18
  RUN git clone --depth=1 https://github.com/OpenAccess-AI-Collective/axolotl.git
19
  # If AXOLOTL_EXTRAS is set, append it in brackets
20
  RUN cd axolotl && \
 
11
 
12
  WORKDIR /workspace
13
 
 
 
 
 
14
  RUN git clone --depth=1 https://github.com/OpenAccess-AI-Collective/axolotl.git
15
  # If AXOLOTL_EXTRAS is set, append it in brackets
16
  RUN cd axolotl && \
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
  peft @ git+https://github.com/huggingface/peft.git
2
  transformers @ git+https://github.com/huggingface/transformers.git
3
  bitsandbytes>=0.39.0
4
- accelerate==0.21.0
5
  addict
6
  fire
7
  PyYAML==6.0
 
1
  peft @ git+https://github.com/huggingface/peft.git
2
  transformers @ git+https://github.com/huggingface/transformers.git
3
  bitsandbytes>=0.39.0
4
+ accelerate @ git+https://github.com/huggingface/accelerate@2a289f6108e77a77a4efffb3f6316bc98538413b
5
  addict
6
  fire
7
  PyYAML==6.0