device=$1

# python src/query_qa.py -m gpt2 -b 256 -n 0 --device "cuda:1"
#python src/query_qa.py -m gpt2-large -b 128 -n 0 --device "cuda:1"
# python src/query_qa.py -m facebook/opt-1.3b -b 128 -n 0 --device $device
# python src/query_qa.py -m facebook/opt-6.7b -b 32 -n 0 -d $device
python src/query_qa.py -m mistralai/Mistral-7B-v0.1 -b 32 -n 0 --device $device


# python src/query_qa_instructions.py -m gpt2 -b 256 -n 0 --device "cuda:1"
# python src/query_qa_instructions.py -m gpt2-large -b 128 -n 0 --device "cuda:1"
# python src/query_qa_instructions.py -m facebook/opt-1.3b -b 128 -n 0 --device $device
# python src/query_qa_instructions.py -m facebook/opt-6.7b -b 32 -n 0 -d $device
python src/query_qa_instructions.py -m mistralai/Mistral-7B-v0.1 -b 32 -n 0 --device $device
