Enable saving in SentenceTransformers by adding get_config_dict (#32)

- Enable saving in SentenceTransformers by adding get_config_dict (7e1ab6246d86fe02a992b477256aa9f12deb4aea)


Co-authored-by: Edward Ross <[email protected]>

custom_st.py

@@ -160,6 +160,9 @@ class Transformer(nn.Module):
         )
         return output
 
+    def get_config_dict(self) -> dict[str, Any]:
+         return {key: self.__dict__[key] for key in self.config_keys}
+
     def save(self, output_path: str, safe_serialization: bool = True) -> None:
         self.auto_model.save_pretrained(output_path, safe_serialization=safe_serialization)
         self.tokenizer.save_pretrained(output_path)