fix use_cache (#19060)

- set `use_cache` to `True` for consistency with other `transformers` models
This commit is contained in:
Younes Belkada 2022-09-16 09:07:02 +02:00 committed by GitHub
parent 0b5c7e4838
commit c8e40d6fa1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -117,7 +117,7 @@ class TrOCRConfig(PretrainedConfig):
classifier_dropout=0.0,
init_std=0.02,
decoder_layerdrop=0.0,
use_cache=False,
use_cache=True,
scale_embedding=False,
use_learned_position_embeddings=True,
layernorm_embedding=True,