Fixing warning: a view of a leaf Variable that requires grad is being used in an in-place operation.
#6
by
tstandley
- opened
modeling_internvl_chat.py
CHANGED
|
@@ -105,7 +105,7 @@ class InternVLChatModel(PreTrainedModel):
|
|
| 105 |
return_dict = return_dict if return_dict is not None else self.config.use_return_dict
|
| 106 |
|
| 107 |
image_flags = image_flags.squeeze(-1)
|
| 108 |
-
input_embeds = self.language_model.get_input_embeddings()(input_ids)
|
| 109 |
|
| 110 |
vit_embeds = self.extract_feature(pixel_values)
|
| 111 |
vit_embeds = vit_embeds[image_flags == 1]
|
|
|
|
| 105 |
return_dict = return_dict if return_dict is not None else self.config.use_return_dict
|
| 106 |
|
| 107 |
image_flags = image_flags.squeeze(-1)
|
| 108 |
+
input_embeds = self.language_model.get_input_embeddings()(input_ids).clone()
|
| 109 |
|
| 110 |
vit_embeds = self.extract_feature(pixel_values)
|
| 111 |
vit_embeds = vit_embeds[image_flags == 1]
|