From 5a7ba955d3ea7831648f1fc797caf4ea354266c4 Mon Sep 17 00:00:00 2001
From: Bram Vanroy <2779410+BramVanroy@users.noreply.github.com>
Date: Wed, 30 Apr 2025 16:18:57 +0200
Subject: [PATCH] Pin TRL (#212)

Fixes #207
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 73214bf..8bb4617 100644
--- a/setup.py
+++ b/setup.py
@@ -66,7 +66,7 @@ _deps = [
     "tensorboard",
     "torch>=2.1.2",
     "transformers>=4.39.3",
-    "trl>=0.9.6",
+    "trl>=0.9.6,<0.13.0",
     "jinja2>=3.0.0",
     "tqdm>=4.64.1",
 ]