From 97e00cf698d4aac6c1ff7c2701df9dd87115b9b9 Mon Sep 17 00:00:00 2001 From: Anh Uong Date: Mon, 30 Sep 2024 19:13:47 -0600 Subject: [PATCH] build(deps): unset hardcoded trl version to get latest updates Signed-off-by: Anh Uong --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 67623503d..2b63b8f54 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -33,7 +33,7 @@ dependencies = [ "sentencepiece>=0.1.99,<0.3", "tokenizers>=0.13.3,<1.0", "tqdm>=4.66.2,<5.0", -"trl==0.9.6", +"trl>=0.9.3,<1.0", "peft>=0.8.0,<0.13", "protobuf>=5.28.0,<6.0.0", "datasets>=2.15.0,<3.0",