diff --git a/pyproject.toml b/pyproject.toml
index a6473d2..ff39ac2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -34,7 +34,7 @@ train = [
     "hf-transfer>=0.1.9",
     "huggingface_hub>=0.30.2",
     "nlpaug>=1.1.11",
-    "optimum>=1.24.0",
+    "optimum>=2.1.0",
     "peft>=0.15.2",
     "pyarrow>=20.0.0",
     "safetensors>=0.4.3",
@@ -58,7 +58,7 @@ full = [
     "hf-transfer>=0.1.9",
     "huggingface_hub>=0.30.2",
     "nlpaug>=1.1.11",
-    "optimum>=1.24.0",
+    "optimum>=2.1.0",
     "peft>=0.15.2",
     "pyarrow>=20.0.0",
     "pytest>=7.4.0",