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': '',
|