SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Don't init TPU device twice (#4916)

Browse Source
This commit is contained in:
Lysandre Debut
2020-06-10 15:53:15 -04:00
committed by GitHub
parent ef2dcdccaa
commit 466aa57a45
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/file_utils.py
View File

@@ -71,9 +71,7 @@ except ImportError:
try: try:
import torch_xla.core.xla_model as xm import torch_xla.core.xla_model as xm # noqa: F401
tpu_device = xm.xla_device()
if _torch_available: if _torch_available:
_torch_tpu_available = True # pylint: disable= _torch_tpu_available = True # pylint: disable=