Merge pull request huggingface#22 from ROCmSoftwarePlatform/fix-night…

…ly-rocm Add tf-nightly-rocm to _is_tf_available check
etemadiamd · Aug 11, 2023 · 0dcfc5c · 0dcfc5c
2 parents 87a6ed1 + d7bb1be
commit 0dcfc5c
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
@@ -73,6 +73,7 @@
                 "tf-nightly",
                 "tf-nightly-cpu",
                 "tf-nightly-gpu",
+                "tf-nightly-rocm",
                 "intel-tensorflow",
                 "intel-tensorflow-avx512",
                 "tensorflow-rocm",