TF CTRL

2020-01-20 16:23:18 -05:00
parent 7511f3dd89
commit 48ac24020d
2 changed files with 104 additions and 82 deletions
--- a/src/transformers/modeling_ctrl.py
+++ b/src/transformers/modeling_ctrl.py
@@ -241,7 +241,6 @@ CTRL_INPUTS_DOCSTRING = r"""
    CTRL_START_DOCSTRING,
 )
 class CTRLModel(CTRLPreTrainedModel):
-
    def __init__(self, config):
        super().__init__(config)
        self.output_hidden_states = config.output_hidden_states
@@ -439,7 +438,6 @@ class CTRLModel(CTRLPreTrainedModel):
    CTRL_START_DOCSTRING,
 )
 class CTRLLMHeadModel(CTRLPreTrainedModel):
-
    def __init__(self, config):
        super().__init__(config)
        self.transformer = CTRLModel(config)