Update modeling_nort5.py

modeling_nort5.py

@@ -387,7 +387,24 @@ class NorT5Model(NorT5PreTrainedModel):
         self.embedding.word_embedding = value
 
     def get_encoder(self):
-        return self.get_encoder_output
+        class EncoderWrapper:
+            def __call__(cls, *args, **kwargs):
+                return cls.forward(*args, **kwargs)
+
+            def forward(
+                cls,
+                input_ids: Optional[torch.Tensor] = None,
+                attention_mask: Optional[torch.Tensor] = None,
+                output_hidden_states: Optional[bool] = None,
+                output_attentions: Optional[bool] = None,
+                return_dict: Optional[bool] = None,
+            ):
+                return_dict = return_dict if return_dict is not None else self.config.use_return_dict
+
+                return self.get_encoder_output(
+                    input_ids, attention_mask, output_hidden_states, output_attentions, return_dict=return_dict
+                )
+        return EncoderWrapper()
 
     def get_decoder(self):
         return self.get_decoder_output