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 && \
|