Yash Sachdeva commited on
Commit
2d565d4
1 Parent(s): be5ca1e
Files changed (1) hide show
  1. Dockerfile +4 -0
Dockerfile CHANGED
@@ -16,6 +16,10 @@ RUN pip install torch
16
 
17
  RUN pip install accelerate
18
 
 
 
 
 
19
  # Install hugging face hub to download llama2 model
20
  #RUN pip install --upgrade huggingface_hub
21
  #RUN CMAKE_ARGS="-DLLAMA_BLAS=ON -DLLAMA_BLAS_VENDOR=OpenBLAS" pip install 'llama-cpp-python[server]' --upgrade --force-reinstall --no-cache-dir
 
16
 
17
  RUN pip install accelerate
18
 
19
+ RUN pip install sentencepiece
20
+ RUN pip install protobuf
21
+ RUN pip install lxml
22
+
23
  # Install hugging face hub to download llama2 model
24
  #RUN pip install --upgrade huggingface_hub
25
  #RUN CMAKE_ARGS="-DLLAMA_BLAS=ON -DLLAMA_BLAS_VENDOR=OpenBLAS" pip install 'llama-cpp-python[server]' --upgrade --force-reinstall --no-cache-dir