dkdaniz commited on
Commit
41fcb94
1 Parent(s): d11eb1b

Update run.sh

Browse files
Files changed (1) hide show
  1. run.sh +5 -2
run.sh CHANGED
@@ -1,6 +1,9 @@
1
  #!/bin/bash
2
 
3
  export TRANSFORMERS_CACHE=/root/.cache
 
 
4
 
5
- exec python3 ./ingest.py --device_type cuda
6
- exec python3 ./run_localGPT.py --device_type cuda
 
 
1
  #!/bin/bash
2
 
3
  export TRANSFORMERS_CACHE=/root/.cache
4
+ export CMAKE_ARGS="-DLLAMA_CUBLAS=on"
5
+ export FORCE_CMAKE=1
6
 
7
+ exec pip install --timeout 100 -r requirements.txt
8
+ exec python ./ingest.py --device_type cuda
9
+ exec python ./run_localGPT.py --device_type cuda