OpenGVLab
/

InternVL-Chat-V1-5

Image-Text-to-Text

feature-extraction

Model card Files Files and versions Metrics Training metrics Community

czczup commited on May 23

Commit

6561410

•

1 Parent(s): e822119

Update modeling_internvl_chat.py

Files changed (1) hide show

modeling_internvl_chat.py +1 -1

modeling_internvl_chat.py CHANGED Viewed

@@ -246,7 +246,7 @@ class InternVLChatModel(PreTrainedModel):
         queries = []
         image_bs = pixel_values.shape[0]
-        print(f'dynamic ViT batch size: {image_bs}, image_counts: {image_counts}')
         for idx, image_count in enumerate(image_counts):
             image_token = IMG_START_TOKEN + IMG_CONTEXT_TOKEN * self.num_image_token * image_count + IMG_END_TOKEN
             question = image_token + '\n' + questions[idx]

         queries = []
         image_bs = pixel_values.shape[0]
+        # print(f'dynamic ViT batch size: {image_bs}, image_counts: {image_counts}')
         for idx, image_count in enumerate(image_counts):
             image_token = IMG_START_TOKEN + IMG_CONTEXT_TOKEN * self.num_image_token * image_count + IMG_END_TOKEN
             question = image_token + '\n' + questions[idx]