Unverified Commit c8e40d6f authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

fix `use_cache` (#19060)

- set `use_cache` to `True` for consistency with other `transformers` models
parent 0b5c7e48
...@@ -117,7 +117,7 @@ class TrOCRConfig(PretrainedConfig): ...@@ -117,7 +117,7 @@ class TrOCRConfig(PretrainedConfig):
classifier_dropout=0.0, classifier_dropout=0.0,
init_std=0.02, init_std=0.02,
decoder_layerdrop=0.0, decoder_layerdrop=0.0,
use_cache=False, use_cache=True,
scale_embedding=False, scale_embedding=False,
use_learned_position_embeddings=True, use_learned_position_embeddings=True,
layernorm_embedding=True, layernorm_embedding=True,
......
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