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",