Luigi commited on
Commit
e85692d
·
1 Parent(s): 931fd49

adjust requirements

Browse files
Files changed (2) hide show
  1. apt.txt +2 -0
  2. requirements.txt +1 -1
apt.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ rustc
2
+ cargo
requirements.txt CHANGED
@@ -5,7 +5,7 @@
5
  wheel
6
  jieba
7
  docopt
8
- llama-cpp-python --no-binary=:all: --global-option=build_ext --global-option="-DLLAMA_USE_CUBLAS=1"
9
  streamlit
10
  duckduckgo_search
11
  gradio
 
5
  wheel
6
  jieba
7
  docopt
8
+ llama-cpp-python --no-binary=:all: --global-option=build_ext --global-option="--cmake-args=-DGGML_CUDA=on"
9
  streamlit
10
  duckduckgo_search
11
  gradio