Update modeling_internvl_chat.py
Browse files
modeling_internvl_chat.py
CHANGED
|
@@ -347,7 +347,7 @@ class InternVLChatModel(PreTrainedModel):
|
|
| 347 |
else:
|
| 348 |
input_embeds = self.language_model.module.get_input_embeddings()(input_ids)
|
| 349 |
|
| 350 |
-
outputs = self.language_model.generate(
|
| 351 |
inputs_embeds=input_embeds,
|
| 352 |
attention_mask=attention_mask,
|
| 353 |
generation_config=generation_config,
|
|
|
|
| 347 |
else:
|
| 348 |
input_embeds = self.language_model.module.get_input_embeddings()(input_ids)
|
| 349 |
|
| 350 |
+
outputs = self.language_model.module.generate(
|
| 351 |
inputs_embeds=input_embeds,
|
| 352 |
attention_mask=attention_mask,
|
| 353 |
generation_config=generation_config,
|