remove --force-reinstall from Dockerfile to ensure correct pytorch version (#492)
Browse files- docker/Dockerfile +1 -1
docker/Dockerfile
CHANGED
@@ -11,7 +11,7 @@ RUN apt-get update && \
|
|
11 |
|
12 |
WORKDIR /workspace
|
13 |
|
14 |
-
RUN pip3 install
|
15 |
RUN git clone --depth=1 https://github.com/OpenAccess-AI-Collective/axolotl.git
|
16 |
# If AXOLOTL_EXTRAS is set, append it in brackets
|
17 |
RUN cd axolotl && \
|
|
|
11 |
|
12 |
WORKDIR /workspace
|
13 |
|
14 |
+
RUN pip3 install "peft @ git+https://github.com/huggingface/peft.git@main"
|
15 |
RUN git clone --depth=1 https://github.com/OpenAccess-AI-Collective/axolotl.git
|
16 |
# If AXOLOTL_EXTRAS is set, append it in brackets
|
17 |
RUN cd axolotl && \
|