diff --git a/pyproject.toml b/pyproject.toml
index 68a4e97..841d7a2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -15,8 +15,8 @@ requires-python = ">=3.10"
 license = {text = "MIT"}
 dependencies = [
     "accelerate",
-    "datasets==3.2.0",
-    "transformers==4.46.0",
+    "datasets==4.0.0",
+    "transformers==4.55.2",
     "torch",
     "sentence_transformers",
     "azure-ai-ml",
@@ -47,7 +47,7 @@ dev = [
 ]
 
 experiment = [
-    "lm-eval==0.4.5",
+    "lm-eval==0.4.9.1",
     "wandb",
 ]