dkdaniz commited on
Commit
e305cfe
1 Parent(s): 6a1e76c

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -36,4 +36,6 @@ RUN python -m pip install --upgrade pip pytest cmake \
36
  # GRADIO_THEME=huggingface \
37
  # SYSTEM=spaces
38
 
 
 
39
  CMD sh -c "chmod -R 777 -c ./run.sh && ./run.sh"
 
36
  # GRADIO_THEME=huggingface \
37
  # SYSTEM=spaces
38
 
39
+ RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on" pip install llama-cpp-python
40
+
41
  CMD sh -c "chmod -R 777 -c ./run.sh && ./run.sh"