ky32 commited on
Commit
5b163c8
·
verified ·
1 Parent(s): b46d5c6

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -1
Dockerfile CHANGED
@@ -4,7 +4,9 @@ FROM python:3.9
4
  COPY ./requirements.txt /code/requirements.txt
5
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
6
 
7
- RUN CMAKE_ARGS="-DLLAMA_CUBLAS=off" pip install llama-cpp-python
 
 
8
 
9
  RUN useradd -m -u 1000 user
10
  USER user
 
4
  COPY ./requirements.txt /code/requirements.txt
5
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
6
 
7
+ # RUN CMAKE_ARGS="-DLLAMA_CUBLAS=off" pip install llama-cpp-python
8
+ RUN CMAKE_ARGS="-DLLAMA_BLAS=ON -DLLAMA_BLAS_VENDOR=OpenBLAS" pip install llama-cpp-python
9
+
10
 
11
  RUN useradd -m -u 1000 user
12
  USER user