Cran-May commited on
Commit
652a9fc
·
1 Parent(s): b0b8255

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -7,6 +7,9 @@ import requests
7
  import time
8
  import sys
9
 
 
 
 
10
  from huggingface_hub import snapshot_download
11
  from llama_cpp import Llama
12
 
@@ -43,7 +46,7 @@ def get_system_tokens(model):
43
 
44
 
45
  repo_name = "maddes8cht/OpenBuddy-openbuddy-falcon-7b-v6-bf16-gguf"
46
- model_name = "ggml-OpenBuddy-openbuddy-falcon-7b-v6-bf16-Q4_0.gguf"
47
 
48
  snapshot_download(repo_id=repo_name, local_dir=".", allow_patterns=model_name)
49
 
 
7
  import time
8
  import sys
9
 
10
+ os.system("pip install --upgrade pip")
11
+ os.system('''CMAKE_ARGS="-DLLAMA_AVX512=ON -DLLAMA_AVX512_VBMI=ON -DLLAMA_AVX512_VNNI=ON -DLLAMA_FP16_VA=ON" pip install llama-cpp-python''')
12
+
13
  from huggingface_hub import snapshot_download
14
  from llama_cpp import Llama
15
 
 
46
 
47
 
48
  repo_name = "maddes8cht/OpenBuddy-openbuddy-falcon-7b-v6-bf16-gguf"
49
+ model_name = "OpenBuddy-openbuddy-falcon-7b-v6-bf16-Q4_K_S.gguf"
50
 
51
  snapshot_download(repo_id=repo_name, local_dir=".", allow_patterns=model_name)
52