Views
No views yet
1>>> from transformers import BertTokenizer, CPTForConditionalGeneration
2
3>>> tokenizer = BertTokenizer.from_pretrained("RUCAIBox/Erya4FT")
4>>> model = CPTForConditionalGeneration.from_pretrained("RUCAIBox/Erya4FT")
5
6>>> input_ids = tokenizer("竖子不足与谋。", return_tensors='pt')
7>>> input_ids.pop("token_type_ids")
8
9>>> pred_ids = model.generate(max_new_tokens=256, **input_ids)
10>>> print(tokenizer.batch_decode(pred_ids, skip_special_tokens=True))
11 ['这 小 子 不 值 得 与 他 商 量 。']