Hugging Face's logo

Spaces:
traversaal-internal
/
Alif-1.0-8B-Model
Sleeping

App Files Community
alishafique commited on
Commit
5be595d
·
verified ·
1 Parent(s): b6feb3b

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -30,7 +30,7 @@ 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" 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
 
 
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