Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -19,7 +19,7 @@ ENV CUDA_DOCKER_ARCH=all
|
|
| 19 |
ENV LLAMA_CUBLAS=1
|
| 20 |
|
| 21 |
# Install depencencies
|
| 22 |
-
RUN python3 -m pip install --no-cache-dir --upgrade pip pytest cmake scikit-build setuptools fastapi uvicorn sse-starlette pydantic-settings starlette-context huggingface-hub==0.14.1
|
| 23 |
|
| 24 |
# Install llama-cpp-python (build with cuda)
|
| 25 |
RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on" pip install --no-cache-dir llama-cpp-python
|
|
|
|
| 19 |
ENV LLAMA_CUBLAS=1
|
| 20 |
|
| 21 |
# Install depencencies
|
| 22 |
+
RUN python3 -m pip install --no-cache-dir --upgrade pip pytest cmake scikit-build setuptools fastapi uvicorn sse-starlette pydantic-settings starlette-context huggingface-hub==0.14.1 flask
|
| 23 |
|
| 24 |
# Install llama-cpp-python (build with cuda)
|
| 25 |
RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on" pip install --no-cache-dir llama-cpp-python
|