Upload folder using huggingface_hub

modeling_internvl_chat.py

@@ -375,7 +375,7 @@ class InternVLChatModel(PreTrainedModel):
         vit_embeds = self.mlp1(vit_embeds)
         return vit_embeds
 
-    def chat(self, tokenizer, pixel_values, question, generation_config, history=None,
+    def chat(self, tokenizer, pixel_values, question, generation_config, history=None, return_history=False,
              IMG_START_TOKEN='<img>', IMG_END_TOKEN='</img>', IMG_CONTEXT_TOKEN='<IMG_CONTEXT>'):
 
         img_context_token_id = tokenizer.convert_tokens_to_ids(IMG_CONTEXT_TOKEN)
@@ -407,7 +407,10 @@ class InternVLChatModel(PreTrainedModel):
         )
         response = tokenizer.batch_decode(generation_output, skip_special_tokens=True)[0]
         history.append((question, response))
-        return response, history
+        if return_history:
+            return response, history
+        else:
+            return response
 
     @torch.no_grad()
     def generate(