Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -22,7 +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 |
|
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
|
|
|
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==5.18.0 huggingface_hub==0.29.1 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
|