Skip to content

[Bug]: Flair not working with latest pytorch version #3671

@JINO-ROHIT

Description

@JINO-ROHIT

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions