[Bug]: Flair not working with latest pytorch version
Describe the bug
Im using torch version 2.6.0 on windows and I get this error "ImportError: cannot import name 'LayoutLMv2FeatureExtractor' from 'transformers'"
To Reproduce
from flair.models import SequenceTagger
Expected behavior
Logs and Stack traces
---------------------------------------------------------------------------
ImportError Traceback (most recent call last)
Cell In[3], line 1
----> 1 from flair.models import SequenceTagger
File \.venv\Lib\site-packages\flair\__init__.py:43
40 # global variable: arrow symbol
41 _arrow = " → "
---> 43 from . import ( # noqa: E402 import after setting device
44 data,
45 models,
46 nn,
47 trainers,
48 visual,
49 )
51 logging.config.dictConfig(
52 {
53 "version": 1,
(...) 65 }
66 )
68 logger = logging.getLogger("flair")
File \.venv\Lib\site-packages\flair\models\__init__.py:1
----> 1 from .entity_linker_model import SpanClassifier
2 from .entity_mention_linking import EntityMentionLinker
3 from .language_model import LanguageModel
File \.venv\Lib\site-packages\flair\models\entity_linker_model.py:11
8 import torch
9 from deprecated.sphinx import deprecated
---> 11 import flair.embeddings
12 import flair.nn
13 from flair.data import Dictionary, Sentence, Span
File g\.venv\Lib\site-packages\flair\embeddings\__init__.py:2
1 # Expose base classses
----> 2 from flair.embeddings.transformer import (
3 TransformerEmbeddings,
4 TransformerJitDocumentEmbeddings,
5 TransformerJitWordEmbeddings,
6 TransformerOnnxDocumentEmbeddings,
7 TransformerOnnxWordEmbeddings,
8 )
10 from .base import Embeddings, ScalarMix
12 # Expose document embedding classes
File \.venv\Lib\site-packages\flair\embeddings\transformer.py:17
15 from packaging.version import Version
16 from torch.jit import ScriptModule
---> 17 from transformers import (
18 CONFIG_MAPPING,
19 AutoConfig,
20 AutoFeatureExtractor,
21 AutoModel,
22 AutoTokenizer,
23 FeatureExtractionMixin,
24 LayoutLMTokenizer,
25 LayoutLMTokenizerFast,
26 LayoutLMv2FeatureExtractor,
27 PretrainedConfig,
28 PreTrainedTokenizer,
29 T5TokenizerFast,
30 )
31 from transformers.tokenization_utils_base import LARGE_INTEGER
32 from transformers.utils import PaddingStrategy
ImportError: cannot import name 'LayoutLMv2FeatureExtractor' from 'transformers' (\.venv\Lib\site-packages\transformers\__init__.py)
Screenshots
No response
Additional Context
No response
Environment
accelerate==1.7.0 annotated-types==0.7.0 asttokens==3.0.0 attrs==25.3.0 beautifulsoup4==4.13.4 bioc==2.1 blis==1.3.0 boto3==1.38.31 botocore==1.38.31 catalogue==2.0.10 certifi==2025.4.26 charset-normalizer==3.4.2 click==8.2.1 cloudpathlib==0.21.1 colorama==0.4.6 comm==0.2.2 confection==0.1.5 conllu==4.5.3 contourpy==1.3.2 cycler==0.12.1 cymem==2.0.11 datasketch==1.6.5 debugpy==1.8.14 decorator==5.2.1 Deprecated==1.2.18 docopt==0.6.2 executing==2.2.0 filelock==3.13.1 flair==0.15.1 fonttools==4.58.2 fsspec==2024.6.1 ftfy==6.3.1 gdown==5.2.0 huggingface-hub==0.32.4 idna==3.10 intervaltree==3.1.0 ipykernel==6.29.5 ipython==9.3.0 ipython_pygments_lexers==1.1.1 ipywidgets==8.1.7 jedi==0.19.2 Jinja2==3.1.4 jmespath==1.0.1 joblib==1.5.1 jsonlines==4.0.0 jupyter_client==8.6.3 jupyter_core==5.8.1 jupyterlab_widgets==3.0.15 kiwisolver==1.4.8 langcodes==3.5.0 langdetect==1.0.9 language_data==1.3.0 lxml==5.4.0 marisa-trie==1.2.1 Markdown==3.8 markdown-it-py==3.0.0 MarkupSafe==2.1.5 matplotlib==3.10.3 matplotlib-inline==0.1.7 mdurl==0.1.2 more-itertools==10.7.0 mpld3==0.5.10 mpmath==1.3.0 murmurhash==1.0.13 nest-asyncio==1.6.0 networkx==3.3 nltk==3.9.1 numpy==2.2.6 packaging==25.0 parso==0.8.4 phonenumbers==8.13.55 pillow==11.0.0 platformdirs==4.3.8 pptree==3.1 preshed==3.0.10 presidio_analyzer==2.2.358 prompt_toolkit==3.0.51 protobuf==6.31.1 psutil==7.0.0 pure_eval==0.2.3 pydantic==2.11.5 pydantic_core==2.33.2 Pygments==2.19.1 pyparsing==3.2.3 PySocks==1.7.1 python-dateutil==2.9.0.post0 pytorch_revgrad==0.2.0 pywin32==310 PyYAML==6.0.2 pyzmq==26.4.0 regex==2024.11.6 requests==2.32.3 requests-file==2.1.0 rich==14.0.0 s3transfer==0.13.0 safetensors==0.5.3 scikit-learn==1.7.0 scipy==1.15.3 segtok==1.5.11 sentencepiece==0.2.0 shellingham==1.5.4 six==1.17.0 smart-open==7.1.0 sortedcontainers==2.4.0 soupsieve==2.7 spacy==3.8.7 spacy-legacy==3.0.12 spacy-loggers==1.0.5 sqlitedict==2.1.0 srsly==2.5.1 stack-data==0.6.3 sympy==1.13.1 tabulate==0.9.0 thinc==8.3.6 threadpoolctl==3.6.0 tldextract==5.3.0 tokenizers==0.21.1 torch==2.6.0 torchaudio==2.7.1+cu128 torchvision==0.22.1+cu128 tornado==6.5.1 tqdm==4.67.1 traitlets==5.14.3 transformer-smaller-training-vocab==0.4.1 transformers==4.52.4 typer==0.16.0 typing-inspection==0.4.1 typing_extensions==4.14.0 urllib3==2.4.0 wasabi==1.1.3 wcwidth==0.2.13 weasel==0.4.1 widgetsnbextension==4.0.14 Wikipedia-API==0.8.1 wrapt==1.17.2