Skip to content

GitLab

Unverified Commit db587220 authored by Arthur's avatar Arthur Committed by GitHub
Browse files

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

nit
parent 51794bf2
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
src/transformers/models/gpt_neo/modeling_gpt_neo.py
View file @ db587220
...@@ -706,7 +706,6 @@ class GPTNeoForCausalLM(GPTNeoPreTrainedModel): ...@@ -706,7 +706,6 @@ class GPTNeoForCausalLM(GPTNeoPreTrainedModel):
model_inputs.update( model_inputs.update(
{ {
"input_ids": input_ids,
"past_key_values": past_key_values, "past_key_values": past_key_values,
"use_cache": kwargs.get("use_cache"), "use_cache": kwargs.get("use_cache"),
"position_ids": position_ids, "position_ids": position_ids,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment