From 14d677ca4a62facf70b28f2922b12e6cd3692a03 Mon Sep 17 00:00:00 2001 From: Julien Plu Date: Thu, 14 Jan 2021 10:35:35 +0100 Subject: [PATCH] Compliancy with tf-nightly (#9570) * Compliancy with tf-nightly * Add more version + restore min version check --- src/transformers/file_utils.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/src/transformers/file_utils.py b/src/transformers/file_utils.py index a194b38b84..f365e469c3 100644 --- a/src/transformers/file_utils.py +++ b/src/transformers/file_utils.py @@ -89,8 +89,20 @@ 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("tensorflow-cpu") except importlib_metadata.PackageNotFoundError: - _tf_version = None - _tf_available = False + try: + _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 version.parse(_tf_version) < version.parse("2"): logger.info(f"TensorFlow found but with version {_tf_version}. Transformers requires version 2 minimum.")