From 63ced828d86dcda4d11258f02492edc18c80aaa9 Mon Sep 17 00:00:00 2001 From: liushz Date: Mon, 11 Sep 2023 10:58:17 +0800 Subject: [PATCH] Update get_started.md (#377) --- docs/en/get_started.md | 4 ++-- docs/zh_cn/get_started.md | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/docs/en/get_started.md b/docs/en/get_started.md index b913b585..74e57fee 100644 --- a/docs/en/get_started.md +++ b/docs/en/get_started.md @@ -127,7 +127,7 @@ For HuggingFace models, users can set model parameters directly through the comm ```bash python run.py --datasets siqa_gen winograd_ppl \ ---hf-model facebook/opt-125m \ +--hf-path facebook/opt-125m \ --model-kwargs device_map='auto' \ --tokenizer-kwargs padding_side='left' truncation='left' trust_remote_code=True \ --max-seq-len 2048 \ @@ -233,7 +233,7 @@ If the HuggingFace model you want to test is not among them, you can also direct ```bash python run.py \ ---hf-model facebook/opt-350m \ # HuggingFace model path +--hf-path facebook/opt-350m \ # HuggingFace model path --tokenizer-path facebook/opt-350m \ # HuggingFace tokenizer path (if the same as the model path, can be omitted) --tokenizer-kwargs padding_side='left' truncation='left' trust_remote_code=True \ # Arguments to construct the tokenizer --model-kwargs device_map='auto' \ # Arguments to construct the model diff --git a/docs/zh_cn/get_started.md b/docs/zh_cn/get_started.md index b8d9a649..768871aa 100644 --- a/docs/zh_cn/get_started.md +++ b/docs/zh_cn/get_started.md @@ -126,7 +126,7 @@ python tools/list_configs.py llama mmlu ```bash python run.py --datasets siqa_gen winograd_ppl \ ---hf-model facebook/opt-125m \ +--hf-path facebook/opt-125m \ --model-kwargs device_map='auto' \ --tokenizer-kwargs padding_side='left' truncation='left' trust_remote_code=True \ --max-seq-len 2048 \ @@ -233,7 +233,7 @@ opt350m = dict( ```bash python run.py \ ---hf-model facebook/opt-350m \ # HuggingFace 模型地址 +--hf-path facebook/opt-350m \ # HuggingFace 模型地址 --tokenizer-path facebook/opt-350m \ # HuggingFace tokenizer 地址(如与模型地址相同,可省略) --tokenizer-kwargs padding_side='left' truncation='left' trust_remote_code=True \ # 构造 tokenizer 的参数 --model-kwargs device_map='auto' \ # 构造 model 的参数