cppowboy commited on
Commit
a0529f8
1 Parent(s): 814b014

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -3
README.md CHANGED
@@ -1,3 +1,8 @@
 
 
 
 
 
1
  # VisCPM
2
  简体中文 | [English](README_en.md)
3
 
@@ -194,9 +199,9 @@ VisCPM需要单卡40GB以上的GPU运行,我们会在尽快更新更加节省
194
  >>> from transformers import AutoModel, AutoTokenizer, AutoImageProcessor
195
  >>> from PIL import Image
196
 
197
- >>> tokenizer = AutoTokenizer.from_pretrained('viscpm', trust_remote_code=True)
198
- >>> processor = AutoImageProcessor.from_pretrained('viscpm', trust_remote_code=True)
199
- >>> model = AutoModel.from_pretrained('viscpm', trust_remote_code=True).to('cuda')
200
 
201
  >>> data = [{
202
  >>> 'context': '',
 
1
+ ---
2
+ language:
3
+ - zh
4
+ - en
5
+ ---
6
  # VisCPM
7
  简体中文 | [English](README_en.md)
8
 
 
199
  >>> from transformers import AutoModel, AutoTokenizer, AutoImageProcessor
200
  >>> from PIL import Image
201
 
202
+ >>> tokenizer = AutoTokenizer.from_pretrained('openbmb/VisCPM-Chat', trust_remote_code=True)
203
+ >>> processor = AutoImageProcessor.from_pretrained('openbmb/VisCPM-Chat', trust_remote_code=True)
204
+ >>> model = AutoModel.from_pretrained('openbmb/VisCPM-Chat', trust_remote_code=True).to('cuda')
205
 
206
  >>> data = [{
207
  >>> 'context': '',