diff --git a/flair/embeddings/transformer.py b/flair/embeddings/transformer.py index d4bf356122..f3492178f9 100644 --- a/flair/embeddings/transformer.py +++ b/flair/embeddings/transformer.py @@ -1073,7 +1073,7 @@ def __init__( model, add_prefix_space=True, **transformers_tokenizer_kwargs, **kwargs ) try: - self.feature_extractor = AutoFeatureExtractor.from_pretrained(model, apply_ocr=False) + self.feature_extractor = AutoFeatureExtractor.from_pretrained(model, apply_ocr=False, **kwargs) except OSError: self.feature_extractor = None else: