riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

[GptNeox] don't gather on pkv when using the trainer (#29892)

Browse Source 
don't gather on pkv when using the trainer
This commit is contained in:
Arthur 2024-03-28 16:56:53 +09:00 committed by GitHub
parent b256516a8c
commit 543889f3f6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/transformers/models/gpt_neox/configuration_gpt_neox.py
View File

@ -103,6 +103,7 @@ class GPTNeoXConfig(PretrainedConfig):
```"""
model_type = "gpt_neox"
keys_to_ignore_at_inference = ["past_key_values"]
def __init__(
self,