Update Dockerfile
Browse files- 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
|