| from transformers import PretrainedConfig | |
| import torch | |
| class CybersecurityKnowledgeGraphConfig(PretrainedConfig): | |
| def __init__( | |
| self, | |
| event_nugget_model_path : str = "nugget_model_state_dict.pth", | |
| event_argument_model_path : str = "argument_model_state_dict.pth", | |
| event_realis_model_path : str = "realis_model_state_dict.pth", | |
| **kwargs, | |
| ): | |
| self.event_nugget_model_path = event_nugget_model_path | |
| self.event_argument_model_path = event_argument_model_path | |
| self.event_realis_model_path = event_realis_model_path | |
| super().__init__(**kwargs) |