dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update

This commit is contained in:
Arthur Zucker 2024-05-28 11:43:32 +02:00
parent 9d62ba5f53
commit d1bc03bfd7
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/transformers/models/gemma/configuration_gemma.py
View File

@ -87,6 +87,7 @@ class GemmaConfig(PretrainedConfig):
>>> configuration = model.config
```"""
model_type = "gemma"
keys_to_ignore_at_inference = ["past_key_values"]
def __init__(
self,

1
src/transformers/models/gemma/diff_gemma.py
View File

@ -111,6 +111,7 @@ class GemmaConfig(PretrainedConfig):
>>> configuration = model.config
```"""
model_type = "gemma"
keys_to_ignore_at_inference = ["past_key_values"]
def __init__(
self,