remove --force-reinstall from Dockerfile to ensure correct pytorch version (#492)

This commit is contained in:
Aman Gupta Karmani
2023-08-29 09:17:51 -04:00
committed by GitHub
parent 48c56470d0
commit e356b297cb

View File

@@ -11,7 +11,7 @@ RUN apt-get update && \
WORKDIR /workspace
RUN pip3 install --force-reinstall "peft @ git+https://github.com/huggingface/peft.git@main"
RUN pip3 install "peft @ git+https://github.com/huggingface/peft.git@main"
RUN git clone --depth=1 https://github.com/OpenAccess-AI-Collective/axolotl.git
# If AXOLOTL_EXTRAS is set, append it in brackets
RUN cd axolotl && \