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')
|