| 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) |
