Upload model

configuration.py

@@ -1,7 +1,7 @@
 from transformers import PretrainedConfig
 import torch
 
-import cybersecurity_knowledge_graph.utils
+from . import utils
 
 
 
@@ -17,9 +17,9 @@ class CybersecurityKnowledgeGraphConfig(PretrainedConfig):
         self.event_argument_model_path = event_argument_model_path
         self.event_realis_model_path = event_realis_model_path
 
-        self.event_nugget_list = cybersecurity_knowledge_graph.utils.event_nugget_list
-        self.event_args_list = cybersecurity_knowledge_graph.utils.event_args_list
-        self.realis_list = cybersecurity_knowledge_graph.utils.realis_list
-        self.arg_2_role = cybersecurity_knowledge_graph.utils.arg_2_role
+        self.event_nugget_list = utils.event_nugget_list
+        self.event_args_list = utils.event_args_list
+        self.realis_list = utils.realis_list
+        self.arg_2_role = utils.arg_2_role
 
         super().__init__(**kwargs)