Update README.md
Browse files
README.md
CHANGED
|
@@ -17,9 +17,10 @@ PromptCLUE:全中文任务零样本学习模型
|
|
| 17 |
加载模型:
|
| 18 |
|
| 19 |
```python
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
|
|
|
| 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 |
使用模型进行预测推理方法:
|