Upload model

configuration.py

@@ -1,7 +1,8 @@
 from transformers import PretrainedConfig
 import torch
 
-from utils import event_args_list, event_nugget_list, realis_list, arg_2_role
+import utils
+
 
 
 class CybersecurityKnowledgeGraphConfig(PretrainedConfig):
@@ -18,9 +19,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 = event_nugget_list
-        self.event_args_list = event_args_list
-        self.realis_list = realis_list
-        self.arg_2_role = 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)