doc and conversion

2019-10-09 17:06:30 +02:00
parent c19b8e4ae0
commit 48b438ff2a
3 changed files with 18 additions and 4 deletions
--- a/transformers/init.py
+++ b/transformers/init.py
@@ -155,6 +155,11 @@ if is_tf_available():
                                         load_distilbert_pt_weights_in_tf2,
                                         TF_DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP)

+    from .modeling_tf_ctrl import (TFCTRLPreTrainedModel, TFCTRLModel,
+                                    TFCTRLLMHeadModel,
+                                    load_ctrl_pt_weights_in_tf2,
+                                    TF_CTRL_PRETRAINED_MODEL_ARCHIVE_MAP)
+
 # TF 2.0 <=> PyTorch conversion utilities
 if is_tf_available() and is_torch_available():
    from .modeling_tf_pytorch_utils import (convert_tf_weight_name_to_pt_weight_name,