Fixing warning: a view of a leaf Variable that requires grad is being used in an in-place operation.

#6
Files changed (1) hide show
  1. modeling_internvl_chat.py +1 -1
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]