Added more dependencies;

2024-10-27 10:06:37 +01:00
parent 936756047f
commit 6906e9314e
1 changed files with 1 additions and 0 deletions
--- a/1
+++ b/1
@@ -28,6 +28,7 @@ WORKDIR /workspace
 RUN pip install trl
 RUN pip install peft
 RUN pip install transformers datasets huggingface-hub scipy
+RUN pip install grokadamw schedulefree

 RUN git clone https://github.com/ggerganov/llama.cpp.git
 WORKDIR /workspace/llama.cpp