File size: 414 Bytes
b0c0df0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
TASK=open_asr_tedlium
CKPT_PATH=lmms-lab/Aero-1-Audio
echo $TASK
TASK_SUFFIX="${TASK//,/_}"
echo $TASK_SUFFIX
accelerate launch --num_processes 8 --main_process_port 30000 -m lmms_eval \
--model aero \
--model_args pretrained=$CKPT_PATH,attn_implementation="flash_attention_2" \
--tasks $TASK \
--batch_size 32 \
--log_samples_suffix $TASK_SUFFIX \
--output_path ./logs/ --verbosity DEBUG
|