add peft install back since it doesn't get installed by setup.py (#331)

2023-07-31 16:31:53 -04:00
parent 6c9a87c8ee
commit db2a3586f3
2 changed files with 12 additions and 7 deletions
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -11,6 +11,7 @@ RUN apt-get update && \

 WORKDIR /workspace

+RUN pip3 install --force-reinstall "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 && \