Update README.md
Browse files
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('
|
198 |
-
>>> processor = AutoImageProcessor.from_pretrained('
|
199 |
-
>>> model = AutoModel.from_pretrained('
|
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': '',
|