[GPTNeo] Add input_embeds functionality to gpt_neo Causal LM (huggi…

…ngface#25664) nit
parambharat · Sep 26, 2023 · 97bbc5d · 97bbc5d
1 parent 31f960a
commit 97bbc5d
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,