radna
/

mini_intern_chat_triton

Visual Question Answering

feature-extraction

Model card Files Files and versions Community

radna commited on Jun 29

Commit

664bae3

•

1 Parent(s): 01488b9

Update modeling_internvl_chat.py

Files changed (1) hide show

modeling_internvl_chat.py +1 -1

modeling_internvl_chat.py CHANGED Viewed

@@ -347,7 +347,7 @@ class InternVLChatModel(PreTrainedModel):
         else:
             input_embeds = self.language_model.module.get_input_embeddings()(input_ids)
-        outputs = self.language_model.generate(
             inputs_embeds=input_embeds,
             attention_mask=attention_mask,
             generation_config=generation_config,

         else:
             input_embeds = self.language_model.module.get_input_embeddings()(input_ids)
+        outputs = self.language_model.module.generate(
             inputs_embeds=input_embeds,
             attention_mask=attention_mask,
             generation_config=generation_config,