OpenGVLab
/

Mini-InternVL2-4B-DA-RS

Image-Text-to-Text

feature-extraction

Model card Files Files and versions Community

zwgao commited on 7 days ago

Commit

f685c6b

•

1 Parent(s): 9e7c039

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -160,7 +160,7 @@ def load_image(image_file, input_size=448, max_num=12):
     return pixel_values
 # If you want to load a model using multiple GPUs, please refer to the `Multiple GPUs` section.
-path = 'OpenGVLab/InternVL2-4B'
 model = AutoModel.from_pretrained(
     path,
     torch_dtype=torch.bfloat16,

     return pixel_values
 # If you want to load a model using multiple GPUs, please refer to the `Multiple GPUs` section.
+path = 'OpenGVLab/Mini-InternVL2-4B-DA-RS'
 model = AutoModel.from_pretrained(
     path,
     torch_dtype=torch.bfloat16,