phongtran commited on
Commit
06ea1e7
·
1 Parent(s): e52b734
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -20,7 +20,7 @@ ENV LLAMA_CUBLAS=1
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
 
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 transformers
24
 
25
  # Install llama-cpp-python (build with cuda)
26
  RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on" pip install llama-cpp-python