alishafique commited on
Commit
22bb20b
·
verified ·
1 Parent(s): 5be595d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -3
Dockerfile CHANGED
@@ -30,9 +30,8 @@ RUN python3 -m pip install --upgrade pip pytest cmake \
30
  #RUN python3 -m pip install llama-cpp-python \
31
  # --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121
32
 
33
- RUN CMAKE_ARGS="-DGGML_CUDA=on -DCMAKE_CUDA_ARCHITECTURES=75" FORCE_CMAKE=1 python3 -m pip install llama-cpp-python \
34
- --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121 \
35
- --force-reinstall --no-cache-dir --verbose
36
 
37
 
38
  RUN useradd -m -u 1000 user
 
30
  #RUN python3 -m pip install llama-cpp-python \
31
  # --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu121
32
 
33
+ RUN pip install https://github.com/abetlen/llama-cpp-python/releases/download/v0.3.4-cu121/llama_cpp_python-0.3.4-cp310-cp310-linux_x86_64.whl
34
+
 
35
 
36
 
37
  RUN useradd -m -u 1000 user