fix self.config.return_dict

modeling_emcoder.py

@@ -200,7 +200,7 @@ class EmCoder(PreTrainedModel):
         Returns:
             Logits of shape (n_samples, B, num_labels).
         """
-        return_dict = return_dict if return_dict is not None else self.config.use_return_dict
+        return_dict = return_dict if return_dict is not None else True
 
         x = input_ids if input_ids is not None else kwargs.get("x")
         mask = attention_mask if attention_mask is not None else kwargs.get("mask")
@@ -264,7 +264,7 @@ class EmCoder(PreTrainedModel):
         **kwargs,
     ) -> tuple[torch.Tensor, ...] | SequenceClassifierOutput:
         """Standard forward pass without MC Dropout."""
-        return_dict = return_dict if return_dict is not None else self.config.use_return_dict
+        return_dict = return_dict if return_dict is not None else True
 
         x = input_ids if input_ids is not None else kwargs.get("x")
         mask = attention_mask if attention_mask is not None else kwargs.get("mask")