File size: 377 Bytes
c95c7b0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | case $1 in
gptj)
MODEL=EleutherAI/gpt-j-6B
;;
mgpt)
MODEL=sberbank-ai/mGPT
;;
*)
MODEL=$1
esac
task_no_slash=$(basename "$2")
\time -o ../outputs/time-$1.$task_no_slash.txt -f "%E" python3 -m main \
--model_api_name 'hf-causal' \
--model_args pretrained=$MODEL \
--task_name $2 \
--num_fewshot 0 \
--use_cache \
--device cuda \
--batch_size $3
|