radna
/

mini_intern_chat_triton

Visual Question Answering

feature-extraction

Model card Files Files and versions

opengvlab-admin commited on May 28, 2024

Commit

58134b8

·

verified ·

1 Parent(s): 5e72178

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -151,7 +151,7 @@ def load_image(image_file, input_size=448, max_num=6):
     return pixel_values
-path = "OpenGVLab/Mini-InternVL-Chat-2B-V1-5"
 model = AutoModel.from_pretrained(
     path,
     torch_dtype=torch.bfloat16,

     return pixel_values
+path = "OpenGVLab/Mini-InternVL-Chat-4B-V1-5"
 model = AutoModel.from_pretrained(
     path,
     torch_dtype=torch.bfloat16,