From 300492ae474e2e06d7e27f82c4781baf9a6595a2 Mon Sep 17 00:00:00 2001
From: Arjun Krishnakumar <arjun.krishnakumar.mec@gmail.com>
Date: Thu, 27 Aug 2026 07:51:09 +0200
Subject: [PATCH] chore: update dependencies

---
 pyproject.toml | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index b2e0d43..21a9b34 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -17,17 +17,19 @@ dependencies = [
 
 [project.optional-dependencies]
 trl = [
-    "torch==2.9.1+cu126",
-    "trl",
-    "deepspeed",
-    "transformers",
-    "datasets>4.5.0",
-    "accelerate",
-    "kernels",
+    "accelerate==1.14.0",
+    "datasets==5.0.0",
+    "deepspeed==0.19.2",
     "flash_attn",
-    "liger-kernel",
+    "huggingface_hub==1.21.0",
+    "liger_kernel==0.7.0",
+    "torch==2.9.1+cu126",
     "tensorboard",
+    "transformers==5.3.0",
+    "triton==3.5.1",
+    "trl==1.7.0",
 ]
+
 flash-attn-2 = ["flash-attn"]
 dev = ["pre-commit", "pytest"]