alishafique commited on
Commit
bdce00b
·
verified ·
1 Parent(s): 34d5793

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -6,7 +6,8 @@ ENV HOST 0.0.0.0
6
  # Install dependencies
7
  RUN python3 -m pip install --upgrade pip pytest cmake \
8
  scikit-build setuptools fastapi uvicorn sse-starlette \
9
- pydantic-settings starlette-context gradio huggingface_hub hf_transfer
 
10
 
11
  # Install llama-cpp-python for CPU
12
  RUN python3 -m pip install llama-cpp-python
 
6
  # Install dependencies
7
  RUN python3 -m pip install --upgrade pip pytest cmake \
8
  scikit-build setuptools fastapi uvicorn sse-starlette \
9
+ pydantic-settings starlette-context gradio huggingface_hub hf_transfer \
10
+ transformers==4.38.2 accelerate bitsandbytes optimum
11
 
12
  # Install llama-cpp-python for CPU
13
  RUN python3 -m pip install llama-cpp-python