Update Dockerfile
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
@@ -24,7 +24,8 @@ COPY requirements.txt .
|
|
24 |
# Install Python dependencies
|
25 |
RUN pip3 install --no-cache-dir --upgrade pip && \
|
26 |
pip3 install --no-cache-dir -r requirements.txt
|
27 |
-
|
|
|
28 |
# Install llama-cpp-python with CUDA support
|
29 |
#ENV FORCE_CMAKE=1
|
30 |
#ENV CMAKE_ARGS="-DGGML_CUDA=on"
|
|
|
24 |
# Install Python dependencies
|
25 |
RUN pip3 install --no-cache-dir --upgrade pip && \
|
26 |
pip3 install --no-cache-dir -r requirements.txt
|
27 |
+
|
28 |
+
RUN pip3 install llama-cpp-python==0.1.0+cu121
|
29 |
# Install llama-cpp-python with CUDA support
|
30 |
#ENV FORCE_CMAKE=1
|
31 |
#ENV CMAKE_ARGS="-DGGML_CUDA=on"
|