From 466aa57a45bfb9fc47d4b75d22c02c34b4b4b0fc Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Wed, 10 Jun 2020 15:53:15 -0400 Subject: [PATCH] Don't init TPU device twice (#4916) --- src/transformers/file_utils.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/transformers/file_utils.py b/src/transformers/file_utils.py index a6925aa082..433c77ae5a 100644 --- 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=