Unverified Commit 5c0bf397 authored by Funtowicz Morgan's avatar Funtowicz Morgan Committed by GitHub
Browse files

Add support for detecting intel-tensorflow version (#10781)


Signed-off-by: default avatarMorgan Funtowicz <funtowiczmo@gmail.com>
parent 0282e24e
......@@ -102,8 +102,12 @@ if USE_TF in ENV_VARS_TRUE_AND_AUTO_VALUES and USE_TORCH not in ENV_VARS_TRUE_VA
try:
_tf_version = importlib_metadata.version("tf-nightly-gpu")
except importlib_metadata.PackageNotFoundError:
_tf_version = None
_tf_available = False
# Support for intel-tensorflow version
try:
_tf_version = importlib_metadata.version("intel-tensorflow")
except importlib_metadata.PackageNotFoundError:
_tf_version = None
_tf_available = False
if _tf_available:
if version.parse(_tf_version) < version.parse("2"):
logger.info(f"TensorFlow found but with version {_tf_version}. Transformers requires version 2 minimum.")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment