radames commited on
Commit
b670baa
1 Parent(s): ae0a725

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -2
Dockerfile CHANGED
@@ -18,10 +18,12 @@ ENV CUDA_DOCKER_ARCH=all
18
  ENV LLAMA_CUBLAS=1
19
 
20
  # Install depencencies
21
- RUN python3 -m pip install --upgrade pip pytest cmake scikit-build setuptools fastapi uvicorn sse-starlette pydantic-settings gradio huggingface_hub hf_transfer
 
 
22
 
23
  # Install llama-cpp-python (build with cuda)
24
- RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on" FORCE_CMAKE=1 pip install llama-cpp-python
25
 
26
  RUN useradd -m -u 1000 user
27
  # Switch to the "user" user
 
18
  ENV LLAMA_CUBLAS=1
19
 
20
  # Install depencencies
21
+ RUN python3 -m pip install --upgrade pip pytest cmake \
22
+ scikit-build setuptools fastapi uvicorn sse-starlette \
23
+ pydantic-settings starlette-context gradio huggingface_hub hf_transfer
24
 
25
  # Install llama-cpp-python (build with cuda)
26
+ RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on" pip install llama-cpp-python
27
 
28
  RUN useradd -m -u 1000 user
29
  # Switch to the "user" user