Axcel1 commited on
Commit
48cbb24
·
verified ·
1 Parent(s): 43fa323

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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="-DLLAMA_CUBLAS=on"
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 .