From d7bb1bee5c93e045d3167dc577df2c6cbf8b674e Mon Sep 17 00:00:00 2001
From: Jason Furmanek <Jason.Furmanek@amd.com>
Date: Fri, 11 Aug 2023 13:17:35 +0000
Subject: [PATCH] Add tf-nightly-rocm to _is_tf_available check

---
 src/transformers/utils/import_utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/transformers/utils/import_utils.py b/src/transformers/utils/import_utils.py
index 6fb76385da4f93..21f8ef0260fe1c 100644
--- 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",