Merge pull request #3539 from diego-morientez/patch-1

Update transformer.py
flairNLP · Aug 30, 2024 · c674212 · c674212
2 parents d4a48a1 + 14a6502
commit c674212
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: