Update modeling_internvl_chat.py

modeling_internvl_chat.py

@@ -277,16 +277,15 @@ class InternVLChatModel(PreTrainedModel):
             image_tokens = IMG_START_TOKEN + IMG_CONTEXT_TOKEN * self.num_image_token * num_patches + IMG_END_TOKEN
             query = query.replace('<image>', image_tokens, 1)
 
-        print(self.num_image_token * num_patches)
-        print(query)
-
         model_inputs = tokenizer(query, return_tensors='pt')
         input_ids = model_inputs['input_ids'].cuda()
         attention_mask = model_inputs['attention_mask'].cuda()
 
-        print(input_ids.shape)
-        print(attention_mask.shape)
-        
+        if verbose:
+            print(f"hehe: {self.num_image_token * num_patches}")
+            print(f"hehe: {query}")
+            print(f"hehe: {input_ids.shape}")
+            print(f"hehe: {attention_mask.shape}")        
         
         generation_config['eos_token_id'] = eos_token_id
         generation_output = self.generate(