Upload ModularStarEncoder

modularStarEncoder.py

@@ -1,6 +1,6 @@
 from transformers import  Starcoder2Model
 import sys
-from .config import ModularStarEncoderConfig
+from config import ModularStarEncoderConfig
 import os
 from dataclasses import dataclass
 from typing import Optional, Tuple, Union, List
@@ -298,7 +298,6 @@ class ModularStarEncoder(StarEncoder2PreTrainedModel):
 
 
         """
-        return_dict = return_dict if return_dict is not None else self.config.use_return_dict
 
         source_embedding = self.starEncoder2(
                 input_ids,
@@ -307,7 +306,7 @@ class ModularStarEncoder(StarEncoder2PreTrainedModel):
                 inputs_embeds=inputs_embeds,
                 output_attentions=output_attentions,
                 output_hidden_states=True,
-                return_dict=return_dict,
+                return_dict=True,
             ).hidden_states