Fix rl requirement

ProbablyFaiz · Aug 15, 2024 · 4758d40 · 4758d40
1 parent c443c57
commit 4758d40
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -41,7 +41,7 @@ sherlock = [
     "StrEnum",
 ]
 local_llm = [
-    "rl[llm]",
+    "rl[llm] @ git+https://github.com/ProbablyFaiz/rl.git@main",
     "vllm==0.5.3post1; platform_system == 'Linux'",
     "peft",
     "trl",