fewshot.sh 597 Bytes
Newer Older
Israel Abebe Azime's avatar
Israel Abebe Azime committed
1
2
lm_eval --model hf \
        --model_args pretrained=masakhane/African-ultrachat-alpaca  \
JessicaOjo's avatar
JessicaOjo committed
3
        --tasks afrimmlu_direct_amh,afrimmlu_direct_eng,afrimmlu_direct_ewe,afrimmlu_direct_fra,afrimmlu_direct_hau,afrimmlu_direct_ibo,afrimmlu_direct_kin,afrimmlu_direct_lin,afrimmlu_direct_lug,afrimmlu_direct_orm,afrimmlu_direct_sna,afrimmlu_direct_sot,afrimmlu_direct_twi,afrimmlu_direct_wol,afrimmlu_direct_xho,afrimmlu_direct_yor,afrimmlu_direct_zul   \
Israel Abebe Azime's avatar
Israel Abebe Azime committed
4
5
        --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