Update README.md
Browse files
README.md
CHANGED
|
@@ -8,7 +8,7 @@ datasets:
|
|
| 8 |
|
| 9 |
- First model of VLM series (**V**ortex **L**anguage **M**odel)
|
| 10 |
|
| 11 |
-
##
|
| 12 |
- Open [Google Colab](https://colab.research.google.com/)
|
| 13 |
- Create new notebook
|
| 14 |
- Paste this code in the cell:
|
|
@@ -17,7 +17,7 @@ datasets:
|
|
| 17 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 18 |
import torch
|
| 19 |
|
| 20 |
-
model_id = "PingVortex/VLM-1-
|
| 21 |
|
| 22 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 23 |
model = AutoModelForCausalLM.from_pretrained(model_id)
|
|
|
|
| 8 |
|
| 9 |
- First model of VLM series (**V**ortex **L**anguage **M**odel)
|
| 10 |
|
| 11 |
+
## Use the model:
|
| 12 |
- Open [Google Colab](https://colab.research.google.com/)
|
| 13 |
- Create new notebook
|
| 14 |
- Paste this code in the cell:
|
|
|
|
| 17 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 18 |
import torch
|
| 19 |
|
| 20 |
+
model_id = "PingVortex/VLM-1-K1"
|
| 21 |
|
| 22 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 23 |
model = AutoModelForCausalLM.from_pretrained(model_id)
|