Skip to content

Commit 9178c4e

Browse files
committed
WIP: add DAMO-NLP-MT/polylm-13b
1 parent 6bf5072 commit 9178c4e

1 file changed

Lines changed: 16 additions & 0 deletions

File tree

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
#!/bin/bash
2+
3+
PRETRAINED="DAMO-NLP-MT/polylm-13b"
4+
TOKENIZER="DAMO-NLP-MT/polylm-13b,use_fast=False"
5+
MODEL_ARGS="pretrained=${PRETRAINED},tokenizer=${TOKENIZER}"
6+
TASK="jcommonsenseqa-1.1-0.2,jnli-1.1-0.2,marc_ja-1.1-0.2,jsquad-1.1-0.2,jaqket_v2-0.1-0.2,xlsum_ja,xwinograd_ja,mgsm"
7+
NUM_FEWSHOT="3,3,3,2,1,1,0,5"
8+
OUTPUT_PATH="models/DAMO-NLP-MT/polylm-13b/result.json"
9+
python main.py \
10+
--model hf-causal \
11+
--model_args $MODEL_ARGS \
12+
--tasks $TASK \
13+
--num_fewshot $NUM_FEWSHOT \
14+
--device "cuda" \
15+
--no_cache \
16+
--output_path $OUTPUT_PATH

0 commit comments

Comments
 (0)