fix #950

3a126e73 · thomwolf · 7223886d · 3a126e73
Commit 3a126e73 authored Aug 05, 2019 by thomwolf
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 5 deletions

pytorch_transformers/convert_transfo_xl_checkpoint_to_pytorch.py ..._transformers/convert_transfo_xl_checkpoint_to_pytorch.py +4 -5

No files found.
--- a/pytorch_transformers/convert_transfo_xl_checkpoint_to_pytorch.py
+++ b/pytorch_transformers/convert_transfo_xl_checkpoint_to_pytorch.py
@@ -24,10 +24,9 @@ from io import open
 import torch

 import pytorch_transformers.tokenization_transfo_xl as data_utils
-from pytorch_transformers.modeling_transfo_xl import (CONFIG_NAME,
-                                                         WEIGHTS_NAME,
-                                                         TransfoXLConfig,
-                                                         TransfoXLLMHeadModel,
+
+from pytorch_transformers import CONFIG_NAME, WEIGHTS_NAME
+from pytorch_transformers.modeling_transfo_xl import (TransfoXLConfig, TransfoXLLMHeadModel,
                                                      load_tf_weights_in_transfo_xl)
 from pytorch_transformers.tokenization_transfo_xl import (CORPUS_NAME, VOCAB_FILES_NAMES)