diff --git a/poetry.lock b/poetry.lock
index 9e19818..dbac9fe 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -57,6 +57,17 @@ files = [
     {file = "annotated_types-0.7.0.tar.gz", hash = "sha256:aff07c09a53a08bc8cfccb9c85b05f1aa9a2a6f23728d790723543408344ce89"},
 ]
 
+[[package]]
+name = "antlr4-python3-runtime"
+version = "4.9.3"
+description = "ANTLR 4.9.3 runtime for Python 3.7"
+optional = false
+python-versions = "*"
+groups = ["main"]
+files = [
+    {file = "antlr4-python3-runtime-4.9.3.tar.gz", hash = "sha256:f224469b4168294902bb1efa80a8bf7855f24c99aef99cbefc1bcd3cce77881b"},
+]
+
 [[package]]
 name = "anyio"
 version = "4.11.0"
@@ -1035,6 +1046,22 @@ files = [
     {file = "nvidia_nvtx_cu12-12.8.90-py3-none-win_amd64.whl", hash = "sha256:619c8304aedc69f02ea82dd244541a83c3d9d40993381b3b590f1adaed3db41e"},
 ]
 
+[[package]]
+name = "omegaconf"
+version = "2.3.0"
+description = "A flexible configuration library"
+optional = false
+python-versions = ">=3.6"
+groups = ["main"]
+files = [
+    {file = "omegaconf-2.3.0-py3-none-any.whl", hash = "sha256:7b4df175cdb08ba400f45cae3bdcae7ba8365db4d165fc65fd04b050ab63b46b"},
+    {file = "omegaconf-2.3.0.tar.gz", hash = "sha256:d5d4b6d29955cc50ad50c46dc269bcd92c6e00f5f90d23ab5fee7bfca4ba4cc7"},
+]
+
+[package.dependencies]
+antlr4-python3-runtime = "==4.9.*"
+PyYAML = ">=5.1.0"
+
 [[package]]
 name = "packaging"
 version = "25.0"
@@ -2159,4 +2186,4 @@ type = ["pytest-mypy"]
 [metadata]
 lock-version = "2.1"
 python-versions = ">=3.10,<3.15"
-content-hash = "59c467e2aa2a23fc44820b7302e0e9d275a49d02c0558b43461032a65519b852"
+content-hash = "e65a22b1639eec4be8df60dfef8b824f4b844e2022358a75fee5f70b6d2bd4eb"
diff --git a/pyproject.toml b/pyproject.toml
index 13d8040..d06ad41 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -15,6 +15,7 @@ dependencies = [
     "torchaudio (>=2.9.0,<3.0.0)",
     "torchvision (>=0.24.0,<0.25.0)",
     "transformers (>=4.57.1,<5.0.0)",
+    "omegaconf (>=2.3.0,<3.0.0)",
 ]
 
 [tool.poetry]
diff --git a/requirements.txt b/requirements.txt
index 9a209f3..9bba2fd 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,7 @@
 accelerate==1.11.0 ; python_version >= "3.10" and python_version < "3.15"
 annotated-doc==0.0.3 ; python_version >= "3.10" and python_version < "3.15"
 annotated-types==0.7.0 ; python_version >= "3.10" and python_version < "3.15"
+antlr4-python3-runtime==4.9.3 ; python_version >= "3.10" and python_version < "3.15"
 anyio==4.11.0 ; python_version >= "3.10" and python_version < "3.15"
 certifi==2025.10.5 ; python_version >= "3.10" and python_version < "3.15"
 charset-normalizer==3.4.4 ; python_version >= "3.10" and python_version < "3.15"
@@ -35,6 +36,7 @@ nvidia-nccl-cu12==2.27.5 ; python_version >= "3.10" and python_version < "3.15"
 nvidia-nvjitlink-cu12==12.8.93 ; python_version >= "3.10" and python_version < "3.15" and platform_system == "Linux" and platform_machine == "x86_64"
 nvidia-nvshmem-cu12==3.3.20 ; python_version >= "3.10" and python_version < "3.15" and platform_system == "Linux" and platform_machine == "x86_64"
 nvidia-nvtx-cu12==12.8.90 ; python_version >= "3.10" and python_version < "3.15" and platform_system == "Linux" and platform_machine == "x86_64"
+omegaconf==2.3.0 ; python_version >= "3.10" and python_version < "3.15"
 packaging==25.0 ; python_version >= "3.10" and python_version < "3.15"
 pillow==12.0.0 ; python_version >= "3.10" and python_version < "3.15"
 psutil==7.1.2 ; python_version >= "3.10" and python_version < "3.15"