-
Notifications
You must be signed in to change notification settings - Fork 36
Open
Description
请问【第一段代码】中default涉及的三个模型,是需要我自己下载到本地,然后放在同项目的checkpoints目录下吗?如果是这样的话,我没有找到模型的下载方式,只找到了引用方式,见【第二段代码】(是ernie的引用方式),所以不太清楚具体是怎么设计的呢?另外还想请问github上下载的代码还有哪些部分是您未上传的吗?
【第一段代码】
parser.add_argument(
"--vote_model_paths",
default='checkpoints/ernie,checkpoints/roberta_wwm_large_fold,checkpoints/roberta_pair',
type=str,
help="Path to pre-trained models",
)
【第二段代码】
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("nghuyong/ernie-1.0")
model = AutoModel.from_pretrained("nghuyong/ernie-1.0")
Metadata
Metadata
Assignees
Labels
No labels