Skip to content

GitLab

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
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 2 deletions
src/transformers/file_utils.py
View file @ 5c0bf397
...@@ -101,6 +101,10 @@ if USE_TF in ENV_VARS_TRUE_AND_AUTO_VALUES and USE_TORCH not in ENV_VARS_TRUE_VA ...@@ -101,6 +101,10 @@ if USE_TF in ENV_VARS_TRUE_AND_AUTO_VALUES and USE_TORCH not in ENV_VARS_TRUE_VA
except importlib_metadata.PackageNotFoundError: except importlib_metadata.PackageNotFoundError:
try: try:
_tf_version = importlib_metadata.version("tf-nightly-gpu") _tf_version = importlib_metadata.version("tf-nightly-gpu")
except importlib_metadata.PackageNotFoundError:
# Support for intel-tensorflow version
try:
_tf_version = importlib_metadata.version("intel-tensorflow")
except importlib_metadata.PackageNotFoundError: except importlib_metadata.PackageNotFoundError:
_tf_version = None _tf_version = None
_tf_available = False _tf_available = False
......
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