Don't init TPU device twice (#4916)

2020-06-10 15:53:15 -04:00
parent ef2dcdccaa
commit 466aa57a45
1 changed files with 1 additions and 3 deletions
--- 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=