From 533d3f93aa55a4044a6e337448957f92f129d490 Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Tue, 9 Jul 2024 10:56:18 +0100 Subject: [PATCH] [CI] Pin transformers version to < 4.42.0 to make vmap happy (#2278) --- .github/unittest/linux_libs/scripts_rlhf/environment.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/unittest/linux_libs/scripts_rlhf/environment.yml b/.github/unittest/linux_libs/scripts_rlhf/environment.yml index 81951f3869c..6b8800a2531 100644 --- a/.github/unittest/linux_libs/scripts_rlhf/environment.yml +++ b/.github/unittest/linux_libs/scripts_rlhf/environment.yml @@ -17,5 +17,5 @@ dependencies: - pyyaml - scipy - hydra-core - - transformers + - transformers<4.42.0 - datasets