Update README.md
Browse files
README.md
CHANGED
@@ -21,7 +21,7 @@ Erya4FT is based on [Erya](https://huggingface.co/RUCAIBox/Erya) and further fin
|
|
21 |
>>> tokenizer = BertTokenizer.from_pretrained("RUCAIBox/Erya4FT")
|
22 |
>>> model = CPTForConditionalGeneration.from_pretrained("RUCAIBox/Erya4FT")
|
23 |
|
24 |
-
>>> input_ids = tokenizer("
|
25 |
>>> input_ids.pop("token_type_ids")
|
26 |
|
27 |
>>> pred_ids = model.generate(max_new_tokens=256, **input_ids)
|
|
|
21 |
>>> tokenizer = BertTokenizer.from_pretrained("RUCAIBox/Erya4FT")
|
22 |
>>> model = CPTForConditionalGeneration.from_pretrained("RUCAIBox/Erya4FT")
|
23 |
|
24 |
+
>>> input_ids = tokenizer("竖子不足与谋。", return_tensors='pt')
|
25 |
>>> input_ids.pop("token_type_ids")
|
26 |
|
27 |
>>> pred_ids = model.generate(max_new_tokens=256, **input_ids)
|