Fix trlx import (#2984)

trlx/main uses now the stable version of ray. We can switch to the main branch.
LAION-AI · Apr 30, 2023 · 218d30f · 218d30f
1 parent eabf67d
commit 218d30f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model/pyproject.toml b/model/pyproject.toml
@@ -18,7 +18,7 @@ dependencies = [
     "fastlangid>=1.0.11",
     "flash_attn>=0.2.8",
     "gdown",
-    "trlx @ git+https://github.com/CarperAI/trlx.git@v0.6.0",
+    "trlx @ git+https://github.com/CarperAI/trlx.git",
     "ninja>=1.11.1",
     "nltk>=3.8.1",
     "numpy>=1.22.4",