Add nn.Module as superclass (#4533)

2020-05-25 12:29:33 -07:00 · 2020-05-25 12:29:33 -07:00 · adab7f8332
parent 8f7c1c7672
commit adab7f8332
1 changed files with 1 additions and 1 deletions
--- a/src/transformers/modeling_mmbt.py
+++ b/src/transformers/modeling_mmbt.py
@ -149,7 +149,7 @@ MMBT_INPUTS_DOCSTRING = r"""    Inputs:
    MMBT_START_DOCSTRING,
    MMBT_INPUTS_DOCSTRING,
 )
-class MMBTModel(ModuleUtilsMixin):
+class MMBTModel(nn.Module, ModuleUtilsMixin):
    r"""
        Outputs: `Tuple` comprising various elements depending on the configuration (config) and inputs:
            **last_hidden_state**: ``torch.FloatTensor`` of shape ``(batch_size, sequence_length, hidden_size)``