/ tests / transformers / version.py
version.py
1  import transformers
2  from packaging.version import Version
3  
4  transformers_version = Version(transformers.__version__)
5  IS_NEW_FEATURE_EXTRACTION_API = transformers_version >= Version("4.27.0")
6  IS_TRANSFORMERS_V5_OR_LATER = transformers_version.major >= 5