[GPTNeo] Add input_embeds functionality to gpt_neo Causal LM (#25664)

nit
huggingface · Aug 23, 2023 · db58722 · db58722
1 parent 51794bf
commit db58722
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/transformers/models/gpt_neo/modeling_gpt_neo.py b/src/transformers/models/gpt_neo/modeling_gpt_neo.py
@@ -706,7 +706,6 @@ def prepare_inputs_for_generation(self, input_ids, past_key_values=None, inputs_
 
         model_inputs.update(
             {
-                "input_ids": input_ids,
                 "past_key_values": past_key_values,
                 "use_cache": kwargs.get("use_cache"),
                 "position_ids": position_ids,