"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "8e5d1619b3e57367701d74647e87b95f8dba5409"
Unverified Commit 14d677ca authored by Julien Plu's avatar Julien Plu Committed by GitHub
Browse files

Compliancy with tf-nightly (#9570)

* Compliancy with tf-nightly

* Add more version + restore min version check
parent 46ed56cf
...@@ -89,8 +89,20 @@ if USE_TF in ENV_VARS_TRUE_AND_AUTO_VALUES and USE_TORCH not in ENV_VARS_TRUE_VA ...@@ -89,8 +89,20 @@ if USE_TF in ENV_VARS_TRUE_AND_AUTO_VALUES and USE_TORCH not in ENV_VARS_TRUE_VA
try: try:
_tf_version = importlib_metadata.version("tensorflow-cpu") _tf_version = importlib_metadata.version("tensorflow-cpu")
except importlib_metadata.PackageNotFoundError: except importlib_metadata.PackageNotFoundError:
_tf_version = None try:
_tf_available = False _tf_version = importlib_metadata.version("tensorflow-gpu")
except importlib_metadata.PackageNotFoundError:
try:
_tf_version = importlib_metadata.version("tf-nightly")
except importlib_metadata.PackageNotFoundError:
try:
_tf_version = importlib_metadata.version("tf-nightly-cpu")
except importlib_metadata.PackageNotFoundError:
try:
_tf_version = importlib_metadata.version("tf-nightly-gpu")
except importlib_metadata.PackageNotFoundError:
_tf_version = None
_tf_available = False
if _tf_available: if _tf_available:
if version.parse(_tf_version) < version.parse("2"): if version.parse(_tf_version) < version.parse("2"):
logger.info(f"TensorFlow found but with version {_tf_version}. Transformers requires version 2 minimum.") 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