From adab7f8332f2027a2f5ec62e8e8bee25fb59765f Mon Sep 17 00:00:00 2001 From: Sho Arora Date: Mon, 25 May 2020 12:29:33 -0700 Subject: [PATCH] Add nn.Module as superclass (#4533) --- src/transformers/modeling_mmbt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/modeling_mmbt.py b/src/transformers/modeling_mmbt.py index 0eddaa72f0..8c403a5f3b 100644 --- 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)``