gorkemgoknar commited on
Commit
c8028e1
1 Parent(s): d3a7867
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -1,3 +1,9 @@
 
 
 
 
 
 
1
  import re, requests, json
2
  import gradio as gr
3
  import random
@@ -6,7 +12,6 @@ from itertools import chain
6
  import asyncio
7
  from llama_cpp import Llama
8
  import datetime
9
- import os
10
 
11
  from transformers import (
12
  StoppingCriteriaList,
 
1
+ import os
2
+
3
+ # we need to compile a OPENBLAS version for cpu
4
+ # Or get it from https://jllllll.github.io/llama-cpp-python-cuBLAS-wheels/
5
+ os.system('CMAKE_ARGS="-DLLAMA_BLAS=ON -DLLAMA_BLAS_VENDOR=OpenBLAS" pip install llama-cpp-python==0.2.11')
6
+
7
  import re, requests, json
8
  import gradio as gr
9
  import random
 
12
  import asyncio
13
  from llama_cpp import Llama
14
  import datetime
 
15
 
16
  from transformers import (
17
  StoppingCriteriaList,