[CI] Pin transformers version to < 4.42.0 to make vmap happy (#2278)

pytorch · Jul 9, 2024 · 533d3f9 · 533d3f9
1 parent a151923
commit 533d3f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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