diff --git a/bak/init_data.py b/bak/init_data.py
index 477e772..9590782 100644
--- a/bak/init_data.py
+++ b/bak/init_data.py
@@ -12,6 +12,7 @@ class BatchDataBase(BaseModel):
     is_validation: bool | None = None
 
     ann_file: str | None = None
+    ann_file_lbs: str | None = None
     img_prefix: str | None = None
     filter_empty_gt: bool | None = None
     update_cache: bool | None = None
diff --git a/db_utils.py b/db_utils.py
index 4afde3b..ff1fd48 100644
--- a/db_utils.py
+++ b/db_utils.py
@@ -19,6 +19,7 @@ class BatchData(Base):
     is_train = Column(Integer)
     is_validation = Column(Integer)
     ann_file = Column(String)
+    ann_file_lbs = Column(String)
     img_prefix = Column(String)
     filter_empty_gt = Column(Integer)
     update_cache = Column(Integer)