forked from agiresearch/OpenAGI
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrun_openagi.sh
32 lines (29 loc) · 824 Bytes
/
run_openagi.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
export TRANSFORMERS_CACHE="YOUR CACHE DIRECTORY TO STORE HUGGINGFACE TRANSFORMERS";
OPENAI_KEY="YOUR OPENAI API KEY";
SEARCHAPI_KEY="YOUR API KEY OF SERPAPI";
DATA_PATH="./openagi_data/";
DEVICE_LIST="0";
EVAL_DEVICE="cuda:1";
LLM_DEVICE="cuda:2";
BATCH_SIZE=5;
LLM_NAME="flan_t5";
TASK="rltf";
SCHEMA="benchmark";
if [ $SCHEMA = "benchmark" ]
then
python openagi_benchmark.py \
--openai_key ${OPENAI_KEY} \
--huggingface_cache ${TRANSFORMERS_CACHE} \
--data_path ${DATA_PATH} \
--device_list ${DEVICE_LIST} \
--eval_device ${EVAL_DEVICE} \
--llm_device ${LLM_DEVICE} \
--batch_size ${BATCH_SIZE} \
--llm_name ${LLM_NAME} \
--task ${TASK}
else
python openagi_opentask.py \
--searchapi_key ${SEARCHAPI_KEY} \
--openai_key ${OPENAI_KEY} \
--device ${EVAL_DEVICE}
fi