Fix TensorFlow pakage check (#26842)

Add tf-nightly-rocm to _is_tf_available check

Fix TensorFlow pakage check (#26842)
Add tf-nightly-rocm to _is_tf_available check
ef42cb62 · jayfurmanek · GitHub · b002353d · ef42cb62
Unverified Commit ef42cb62 authored Oct 17, 2023 by jayfurmanek Committed by GitHub Oct 17, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/transformers/utils/import_utils.py src/transformers/utils/import_utils.py +1 -0

No files found.
--- a/src/transformers/utils/import_utils.py
+++ b/src/transformers/utils/import_utils.py
@@ -168,6 +168,7 @@ else:
                "tf-nightly",
                "tf-nightly-cpu",
                "tf-nightly-gpu",
+                "tf-nightly-rocm",
                "intel-tensorflow",
                "intel-tensorflow-avx512",
                "tensorflow-rocm",