Update README.md

README.md

@@ -51,7 +51,7 @@ pipeline_tag: image-text-to-text
 
 ## Usage
 
-We provide an example code to run `SkyworkVL-38B` using `transformers`
+We provide an example code to run `SkyworkVL-2B` using `transformers`
 
 ### Model Loading
 
@@ -60,7 +60,7 @@ We provide an example code to run `SkyworkVL-38B` using `transformers`
 ```python
 import torch
 from transformers import AutoTokenizer, AutoModel
-path = "Skywork/SkyworkVL-38B"
+path = "Skywork/SkyworkVL-2B"
 model = AutoModel.from_pretrained(
     path,
     torch_dtype=torch.bfloat16,
@@ -74,7 +74,7 @@ model = AutoModel.from_pretrained(
 ```python
 import torch
 from transformers import AutoTokenizer, AutoModel
-path = "Skywork/SkyworkVL-38B"
+path = "Skywork/SkyworkVL-2B"
 model = AutoModel.from_pretrained(
     path,
     torch_dtype=torch.bfloat16,
@@ -196,8 +196,8 @@ def split_model(model_name):
 
     return device_map
 
-path = 'Skywork/SkyworkVL-38B'
-device_map = split_model('SkyworkVL-38B')
+path = 'Skywork/SkyworkVL-2B'
+device_map = split_model('SkyworkVL-2B')
 model = AutoModel.from_pretrained(
     path,
     torch_dtype=torch.bfloat16,