muryshev commited on
Commit
f5dec7b
1 Parent(s): abad466

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -18,7 +18,7 @@ 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 starlette-context
22
 
23
  # Install llama-cpp-python (build with cuda)
24
  RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on" pip install llama-cpp-python
 
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 starlette-context huggingface-hub==0.14.1 flask
22
 
23
  # Install llama-cpp-python (build with cuda)
24
  RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on" pip install llama-cpp-python