赞
踩
RobertaTokenizer,RobertaForMaskedLM
from transformers import RobertaTokenizer,RobertaForMaskedLM
tokenizer = RobertaTokenizer.from_pretrained('microsoft/graphcodebert-base')
model = RobertaForMaskedLM.from_pretrained('microsoft/graphcodebert-base')
input_ids = tokenizer(['Language model is what I need.','I love China'],padding=True,return_tensors='pt')
print(input_ids)
#out = model(**input_ids)
# print(out)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。