Skip to content

Commit f678199

Browse files
authored
Update pred.py
1 parent 4b7f86b commit f678199

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

evaluation/LongBench/pred.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818

1919
def parse_args(args=None):
2020
parser = argparse.ArgumentParser()
21-
parser.add_argument('--model', type=str, default=None, choices=["Qwen2.5-7B-Instruct", "Meta-Llama-3.1-8B-Instruct", "Qwen2.5-7B-CARE", "Llama-3.1-8B-CARE"])
21+
parser.add_argument('--model', type=str, default=None, choices=["Qwen2.5-7B-Instruct", "Meta-Llama-3.1-8B-Instruct", "Qwen2.5-7B-Instruct-CARE", "Qwen2.5-14B-Instruct-CARE", "Llama-3.1-8B-Instruct-CARE"])
2222
parser.add_argument('--e', action='store_true', help="Evaluate on LongBench-E")
2323
return parser.parse_args(args)
2424

0 commit comments

Comments
 (0)