pabloce commited on
Commit
39d332a
·
verified ·
1 Parent(s): e416ee9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from llama_cpp_agent import LlamaCppAgent
7
  from llama_cpp_agent import MessagesFormatterType
8
  from llama_cpp_agent.providers import LlamaCppPythonProvider
9
 
10
- subprocess.run('pip install llama-cpp-python --no-build-isolation --no-cache-dir --force-reinstall --upgrade', env={'CMAKE_ARGS': "-DLLAMA_CUBLAS=ON", 'FORCE_CMAKE': '1'}, shell=True)
11
 
12
  hf_hub_download(repo_id="TheBloke/Mistral-7B-Instruct-v0.2-GGUF", filename="mistral-7b-instruct-v0.2.Q6_K.gguf", local_dir = "./models")
13
 
 
7
  from llama_cpp_agent import MessagesFormatterType
8
  from llama_cpp_agent.providers import LlamaCppPythonProvider
9
 
10
+ subprocess.run('pip install llama-cpp-python --no-build-isolation --no-cache-dir --force-reinstall --upgrade --index-url=https://smartappli.github.io/llama-cpp-python-cuBLAS-wheels/AVX512/cu122', env={'CMAKE_ARGS': "-DLLAMA_CUBLAS=ON"}, shell=True)
11
 
12
  hf_hub_download(repo_id="TheBloke/Mistral-7B-Instruct-v0.2-GGUF", filename="mistral-7b-instruct-v0.2.Q6_K.gguf", local_dir = "./models")
13