From 218d30f20368f1fd648f522c3d6f07a196a001ad Mon Sep 17 00:00:00 2001 From: sanagno Date: Sun, 30 Apr 2023 18:56:47 +0200 Subject: [PATCH] Fix trlx import (#2984) trlx/main uses now the stable version of ray. We can switch to the main branch. --- model/pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model/pyproject.toml b/model/pyproject.toml index d4abd535c0..0676f30f40 100644 --- 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",