Update Dockerfile
Browse files- Dockerfile +2 -3
Dockerfile
CHANGED
@@ -28,9 +28,8 @@ RUN pip3 install --no-cache-dir --upgrade pip && \
|
|
28 |
# Install llama-cpp-python with CUDA support
|
29 |
ENV FORCE_CMAKE=1
|
30 |
ENV CMAKE_ARGS="-DGGML_CUDA=on"
|
31 |
-
RUN pip3 install llama-cpp-python
|
32 |
-
|
33 |
-
# --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121
|
34 |
|
35 |
# Copy the rest of the application code
|
36 |
COPY . .
|
|
|
28 |
# Install llama-cpp-python with CUDA support
|
29 |
ENV FORCE_CMAKE=1
|
30 |
ENV CMAKE_ARGS="-DGGML_CUDA=on"
|
31 |
+
RUN pip3 install llama-cpp-python \
|
32 |
+
--extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121
|
|
|
33 |
|
34 |
# Copy the rest of the application code
|
35 |
COPY . .
|