diff --git a/model/LMConfig.py b/model/LMConfig.py
index f5e0012..505376d 100644
--- a/model/LMConfig.py
+++ b/model/LMConfig.py
@@ -41,6 +41,12 @@ class LMConfig(PretrainedConfig):
             knowledge_num: int = 64*64,
             knowledge_length: int = 8,
             knowledge_dim: int = 128,
+            ####################################################
+            # Triple extraction related configurations
+            ####################################################
+            max_subject_len: int = 8,
+            max_predicate_len: int = 4,
+            max_object_len: int = 8,
             **kwargs,
     ):
         self.dim = dim
@@ -76,4 +82,10 @@ class LMConfig(PretrainedConfig):
         self.knowledge_num = knowledge_num
         self.knowledge_length = knowledge_length
         self.knowledge_dim = knowledge_dim
+        ####################################################
+        # Triple extraction related configurations
+        ####################################################
+        self.max_subject_len = max_subject_len
+        self.max_predicate_len = max_predicate_len
+        self.max_object_len = max_object_len
         super().__init__(**kwargs)