Include tensorflow-aarch64 as a candidate (#18345)

Co-authored-by: Ankur Goyal <ankur@impira.com>

Include tensorflow-aarch64 as a candidate (#18345)
Co-authored-by: Ankur Goyal <ankur@impira.com>
9c336657 · Ankur Goyal · GitHub · b53dab60 · 9c336657
Unverified Commit 9c336657 authored Jul 28, 2022 by Ankur Goyal Committed by GitHub Jul 28, 2022
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
@@ -71,6 +71,7 @@ if USE_TF in ENV_VARS_TRUE_AND_AUTO_VALUES and USE_TORCH not in ENV_VARS_TRUE_VA
            "intel-tensorflow-avx512",
            "tensorflow-rocm",
            "tensorflow-macos",
+            "tensorflow-aarch64",
        )
        _tf_version = None
        # For the metadata, we have to look for both tensorflow and tensorflow-cpu