riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 10:12:23 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

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

Browse Source 
nit
This commit is contained in:
Arthur 2023-08-23 07:49:19 +02:00 committed by GitHub
parent 51794bf21e
commit db58722084
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/transformers/models/gpt_neo/modeling_gpt_neo.py
View File

@ -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,