ClueAI commited on
Commit
efc3942
1 Parent(s): bc457b7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -3
README.md CHANGED
@@ -17,9 +17,10 @@ PromptCLUE:全中文任务零样本学习模型
17
  加载模型:
18
 
19
  ```python
20
- from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
21
- tokenizer = AutoTokenizer.from_pretrained("ClueAI/PromptCLUE-base")
22
- model = AutoModelForSeq2SeqLM.from_pretrained("ClueAI/PromptCLUE-base")
 
23
  ```
24
 
25
  使用模型进行预测推理方法:
 
17
  加载模型:
18
 
19
  ```python
20
+ # 加载模型
21
+ from transformers import T5Tokenizer, T5ForConditionalGeneration
22
+ tokenizer = T5Tokenizer.from_pretrained("ClueAI/PromptCLUE-base")
23
+ model = T5ForConditionalGeneration.from_pretrained("ClueAI/PromptCLUE-base")
24
  ```
25
 
26
  使用模型进行预测推理方法: