Pin TRL

Fixes #207
huggingface · Feb 27, 2025 · b4bd897 · b4bd897
1 parent ae3f44f
commit b4bd897
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -66,7 +66,7 @@
     "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",
 ]