dkdaniz commited on
Commit
e3ccd16
1 Parent(s): 9281d61

Update run.sh

Browse files
Files changed (1) hide show
  1. run.sh +3 -3
run.sh CHANGED
@@ -1,6 +1,6 @@
1
  #!/bin/bash
2
 
3
- export TRANSFORMERS_CACHE=/home/cache/
4
 
5
- exec python ./ingest.py
6
- exec python ./run_localGPT.py
 
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