Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "5e8c8eb5bad1d43212b2e8bf724114fe5cdfd807"
Unverified Commit 0963a250 authored by Calico's avatar Calico Committed by GitHub
Browse files

fix(configuration_llama): add `keys_to_ignore_at_inference` to `LlamaConfig` (#23891)

parent 00f6ba0e
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/models/llama/configuration_llama.py
View file @ 0963a250
...@@ -79,6 +79,7 @@ class LlamaConfig(PretrainedConfig): ...@@ -79,6 +79,7 @@ class LlamaConfig(PretrainedConfig):
>>> configuration = model.config >>> configuration = model.config
```""" ```"""
model_type = "llama" model_type = "llama"
keys_to_ignore_at_inference = ["past_key_values"]
def __init__( def __init__(
self, self,
......
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