Upload ModularStarEncoder

modularStarEncoder.py

@@ -154,8 +154,8 @@ class StarEncoder2LMPredictionHead(nn.Module):
         super().__init__()
         for element in dir(config):
             value = getattr(config, element)  # Get the attribute value
-            if isinstance(value, tuple) or isinstance(value, list):
-                setattr(config, element, value[-1])
+            if (isinstance(value, tuple) or isinstance(value, list)) and len(value)>0:
+                setattr(config, element, value[0])
         self.transform = StarEncoder2PredictionHeadTransform(config)
 
         # The output weights are the same as the input embeddings, but there is