Update setup.py (#1)

huggingface · Jan 24, 2025 · a4bf904 · a4bf904
1 parent 697c119
commit a4bf904
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 @ git+https://github.com/huggingface/trl.git",
     "jinja2>=3.0.0",
     "tqdm>=4.64.1",
 ]