[common attributes] Fix previous commit for transfo-xl

2019-11-11 20:03:19 -05:00
parent 2f17464266
commit 2aef2f0bbc
2 changed files with 3 additions and 2 deletions
--- a/transformers/init.py
+++ b/transformers/init.py
@@ -72,6 +72,7 @@ if is_torch_available():
                                OpenAIGPTLMHeadModel, OpenAIGPTDoubleHeadsModel,
                                load_tf_weights_in_openai_gpt, OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_MAP)
    from .modeling_transfo_xl import (TransfoXLPreTrainedModel, TransfoXLModel, TransfoXLLMHeadModel,
+                                    AdaptiveEmbedding,
                                    load_tf_weights_in_transfo_xl, TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_MAP)
    from .modeling_gpt2 import (GPT2PreTrainedModel, GPT2Model,
                                GPT2LMHeadModel, GPT2DoubleHeadsModel,