Fix TensorFlow pakage check (#26842)

Add tf-nightly-rocm to _is_tf_available check
huggingface · Oct 17, 2023 · ef42cb6 · ef42cb6
1 parent b002353
commit ef42cb6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/transformers/utils/import_utils.py b/src/transformers/utils/import_utils.py
@@ -168,6 +168,7 @@ def _is_package_available(pkg_name: str, return_version: bool = False) -> Union[
                 "tf-nightly",
                 "tf-nightly-cpu",
                 "tf-nightly-gpu",
+                "tf-nightly-rocm",
                 "intel-tensorflow",
                 "intel-tensorflow-avx512",
                 "tensorflow-rocm",