Update README.md
Browse files
README.md
CHANGED
|
@@ -7,7 +7,7 @@ license: creativeml-openrail-m
|
|
| 7 |
|
| 8 |
这个模型是基于1000亿token中文语料上预训练,并且在数百种任务上进行Prompt任务式训练
|
| 9 |
|
| 10 |
-
|
| 11 |
|
| 12 |
```python
|
| 13 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
|
@@ -15,7 +15,7 @@ tokenizer = AutoTokenizer.from_pretrained("ClueAI/PromptCLUE")
|
|
| 15 |
model = AutoModelForSeq2SeqLM.from_pretrained("ClueAI/PromptCLUE")
|
| 16 |
```
|
| 17 |
|
| 18 |
-
|
| 19 |
```python
|
| 20 |
import torch
|
| 21 |
#device = torch.device('cpu')
|
|
|
|
| 7 |
|
| 8 |
这个模型是基于1000亿token中文语料上预训练,并且在数百种任务上进行Prompt任务式训练
|
| 9 |
|
| 10 |
+
加载模型:
|
| 11 |
|
| 12 |
```python
|
| 13 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
|
|
|
| 15 |
model = AutoModelForSeq2SeqLM.from_pretrained("ClueAI/PromptCLUE")
|
| 16 |
```
|
| 17 |
|
| 18 |
+
使用模型进行预测推理方法:
|
| 19 |
```python
|
| 20 |
import torch
|
| 21 |
#device = torch.device('cpu')
|