add intel-tensorflow-avx512 to the candidates (#12751)

c07334c1 · Nathan Zhou · GitHub · 69892649 · c07334c1
Unverified Commit c07334c1 authored Jul 16, 2021 by Nathan Zhou Committed by GitHub Jul 16, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/transformers/file_utils.py src/transformers/file_utils.py +1 -0

No files found.
--- a/src/transformers/file_utils.py
+++ b/src/transformers/file_utils.py
@@ -88,6 +88,7 @@ if USE_TF in ENV_VARS_TRUE_AND_AUTO_VALUES and USE_TORCH not in ENV_VARS_TRUE_VA
            "tf-nightly-cpu",
            "tf-nightly-gpu",
            "intel-tensorflow",
+            "intel-tensorflow-avx512",
            "tensorflow-rocm",
            "tensorflow-macos",
        )