fewshot.sh 478 Bytes
Newer Older
Israel Abebe Azime's avatar
Israel Abebe Azime committed
1
2
3
4
5
lm_eval --model hf \
        --model_args pretrained=masakhane/African-ultrachat-alpaca  \
        --tasks afrimmlu_amh,afrimmlu_eng,afrimmlu_ewe,afrimmlu_fra,afrimmlu_hau,afrimmlu_ibo,afrimmlu_kin,afrimmlu_lin,afrimmlu_lug,afrimmlu_orm,afrimmlu_sna,afrimmlu_sot,afrimmlu_twi,afrimmlu_wol,afrimmlu_xho,afrimmlu_yor,afrimmlu_zul   \
        --device cuda:0     \
        --batch_size 1 \
Israel Abebe Azime's avatar
Israel Abebe Azime committed
6
        --num_fewshot 0 \
Israel Abebe Azime's avatar
Israel Abebe Azime committed
7
8
        --verbosity DEBUG \
        --wandb_args project=afrimmlu