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

nit

[`GPTNeo`] Add input_embeds functionality to gpt_neo Causal LM (#25664)
nit
db587220 · Arthur · GitHub · 51794bf2 · db587220
Unverified Commit db587220 authored Aug 23, 2023 by Arthur Committed by GitHub Aug 23, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

src/transformers/models/gpt_neo/modeling_gpt_neo.py src/transformers/models/gpt_neo/modeling_gpt_neo.py +0 -1

No files found.
--- a/src/transformers/models/gpt_neo/modeling_gpt_neo.py
+++ b/src/transformers/models/gpt_neo/modeling_gpt_neo.py
@@ -706,7 +706,6 @@ class GPTNeoForCausalLM(GPTNeoPreTrainedModel):

        model_inputs.update(
            {
-                "input_ids": input_ids,
                "past_key_values": past_key_values,
                "use_cache": kwargs.get("use_cache"),
                "position_ids": position_ids,