[release] Pin transformers to at least 4.31.0 for llama2 ft(ray-proje…

…ct#37956) Signed-off-by: xwjiang2010 <xwjiang2010@gmail.com>
anoopsaha · Aug 1, 2023 · ba2e406 · ba2e406
1 parent 672ce55
commit ba2e406
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/doc/source/templates/testing/cluster_envs/04_finetuning_llms_with_deepspeed.yaml b/doc/source/templates/testing/cluster_envs/04_finetuning_llms_with_deepspeed.yaml
@@ -12,7 +12,7 @@ python:
     torchaudio==2.0.1,
     deepspeed,
     fairscale,
-    transformers,
+    transformers>=4.31.0,
     dataset,
     accelerate,
     evaluate,

diff --git a/doc/source/templates/testing/docker/04_finetuning_llms_with_deepspeed/requirements.txt b/doc/source/templates/testing/docker/04_finetuning_llms_with_deepspeed/requirements.txt
@@ -3,7 +3,7 @@ torchvision==0.15.1
 torchaudio==2.0.1
 deepspeed
 fairscale
-transformers
+transformers>=4.31.0
 dataset
 accelerate
 evaluate