Spaces:
Sleeping
Sleeping
Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -23,7 +23,7 @@ RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.10 1
|
|
23 |
RUN pip install --upgrade pip
|
24 |
|
25 |
# Set CUDA build flag for llama-cpp-python
|
26 |
-
ENV CMAKE_ARGS="-
|
27 |
|
28 |
# Copy requirements first for layer caching
|
29 |
COPY requirements.txt .
|
|
|
23 |
RUN pip install --upgrade pip
|
24 |
|
25 |
# Set CUDA build flag for llama-cpp-python
|
26 |
+
ENV CMAKE_ARGS="-DGGML_CUDA=on"
|
27 |
|
28 |
# Copy requirements first for layer caching
|
29 |
COPY requirements.txt .
|