Don't init TPU device twice (#4916)

466aa57a · Lysandre Debut · GitHub · ef2dcdcc · 466aa57a
Unverified Commit 466aa57a authored Jun 10, 2020 by Lysandre Debut Committed by GitHub Jun 10, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 3 deletions

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

No files found.
--- a/src/transformers/file_utils.py
+++ b/src/transformers/file_utils.py
@@ -71,9 +71,7 @@ except ImportError:


 try:
-    import torch_xla.core.xla_model as xm
-
-    tpu_device = xm.xla_device()
+    import torch_xla.core.xla_model as xm  # noqa: F401

    if _torch_available:
        _torch_tpu_available = True  # pylint: disable=