Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +4 -2
Dockerfile
CHANGED
|
@@ -18,10 +18,12 @@ ENV CUDA_DOCKER_ARCH=all
|
|
| 18 |
ENV LLAMA_CUBLAS=1
|
| 19 |
|
| 20 |
# Install depencencies
|
| 21 |
-
RUN python3 -m pip install --upgrade pip pytest cmake
|
|
|
|
|
|
|
| 22 |
|
| 23 |
# Install llama-cpp-python (build with cuda)
|
| 24 |
-
RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on"
|
| 25 |
|
| 26 |
RUN useradd -m -u 1000 user
|
| 27 |
# Switch to the "user" user
|
|
|
|
| 18 |
ENV LLAMA_CUBLAS=1
|
| 19 |
|
| 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
|
| 27 |
|
| 28 |
RUN useradd -m -u 1000 user
|
| 29 |
# Switch to the "user" user
|