Skip to content

GitLab

Unverified Commit 864c1dfe authored by Mats Sjöberg's avatar Mats Sjöberg Committed by GitHub
Browse files

Accept tensorflow-rocm package when checking TF availability (#11595)

parent 3e3e41ae
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/file_utils.py
View file @ 864c1dfe
...@@ -88,6 +88,7 @@ if USE_TF in ENV_VARS_TRUE_AND_AUTO_VALUES and USE_TORCH not in ENV_VARS_TRUE_VA ...@@ -88,6 +88,7 @@ if USE_TF in ENV_VARS_TRUE_AND_AUTO_VALUES and USE_TORCH not in ENV_VARS_TRUE_VA
"tf-nightly-cpu", "tf-nightly-cpu",
"tf-nightly-gpu", "tf-nightly-gpu",
"intel-tensorflow", "intel-tensorflow",
"tensorflow-rocm",
) )
_tf_version = None _tf_version = None
# For the metadata, we have to look for both tensorflow and tensorflow-cpu # For the metadata, we have to look for both tensorflow and tensorflow-cpu
......
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