Spaces:
Paused
Paused
Update Dockerfile
Browse files- 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"
|