Sergidev commited on
Commit
5f54a19
·
verified ·
1 Parent(s): 825b732

Attempt 103

Browse files
Files changed (1) hide show
  1. start.py +1 -1
start.py CHANGED
@@ -3,5 +3,5 @@ import subprocess
3
  # commented because the existing llama-cpp-python package was renoved fron requirements.txt
4
  subprocess.run("pip uninstall -y llama-cpp-python", shell=True)
5
 
6
- install_command = "CMAKE_ARGS='-DLLAMA_CUDA=on' pip install llama-cpp-python"
7
  subprocess.run(install_command, shell=True)
 
3
  # commented because the existing llama-cpp-python package was renoved fron requirements.txt
4
  subprocess.run("pip uninstall -y llama-cpp-python", shell=True)
5
 
6
+ install_command = "CMAKE_ARGS='-DLLAMA_CUDA=on -DCMAKE_CUDA_COMPILER=CUDACXX' pip install llama-cpp-python"
7
  subprocess.run(install_command, shell=True)