ClueAI commited on
Commit
3c40686
·
1 Parent(s): 7471c4f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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')