Update Dockerfile
Browse files- Dockerfile +1 -2
Dockerfile
CHANGED
|
@@ -22,8 +22,7 @@ RUN nvcc --version && python3 --version
|
|
| 22 |
# Install depencencies
|
| 23 |
RUN python3 -m pip install --upgrade pip pytest cmake \
|
| 24 |
scikit-build setuptools fastapi uvicorn sse-starlette \
|
| 25 |
-
pydantic-settings starlette-context gradio huggingface_hub hf_transfer
|
| 26 |
-
transformers==4.38.2 accelerate bitsandbytes optimum
|
| 27 |
|
| 28 |
# Install llama-cpp-python (build with cuda)
|
| 29 |
# RUN CMAKE_ARGS="-DGGML_CUDA=on -DCMAKE_CUDA_ARCHITECTURES=75" FORCE_CMAKE=1 python3 -m pip install llama-cpp-python --force-reinstall --upgrade --no-cache-dir --verbose
|
|
|
|
| 22 |
# Install depencencies
|
| 23 |
RUN python3 -m pip install --upgrade pip pytest cmake \
|
| 24 |
scikit-build setuptools fastapi uvicorn sse-starlette \
|
| 25 |
+
pydantic-settings starlette-context gradio huggingface_hub hf_transfer
|
|
|
|
| 26 |
|
| 27 |
# Install llama-cpp-python (build with cuda)
|
| 28 |
# RUN CMAKE_ARGS="-DGGML_CUDA=on -DCMAKE_CUDA_ARCHITECTURES=75" FORCE_CMAKE=1 python3 -m pip install llama-cpp-python --force-reinstall --upgrade --no-cache-dir --verbose
|