Skip to content

Commit 7c74523

Browse files
authored
pin transformers < 5.0 (#1439)
Signed-off-by: Peter St. John <pstjohn@nvidia.com>
1 parent fbcd380 commit 7c74523

7 files changed

Lines changed: 7 additions & 7 deletions

File tree

.devcontainer/recipes/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ torchdata
1212
torchmetrics
1313
tqdm
1414
transformer_engine
15-
transformers
15+
transformers<5.0
1616
typer
1717
wandb
1818
zstandard

bionemo-recipes/models/llama3/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,4 @@ lm-eval # For testing
22
torch
33
torchao!=0.14.0
44
transformer_engine[pytorch]
5-
transformers
5+
transformers<5.0

bionemo-recipes/recipes/esm2_accelerate_te/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,5 +5,5 @@ hydra-core
55
torchao!=0.14.0
66
torchmetrics
77
transformer_engine[pytorch]
8-
transformers
8+
transformers<5.0
99
wandb

bionemo-recipes/recipes/esm2_native_te/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,6 @@ torchdata
88
torchmetrics
99
tqdm
1010
transformer_engine[pytorch]
11-
transformers
11+
transformers<5.0
1212
wandb
1313
nvdlfw_inspect @ git+https://github.com/NVIDIA/nvidia-dlfw-inspect

bionemo-recipes/recipes/esm2_peft_te/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,5 +4,5 @@ torch
44
torchao!=0.14.0
55
tqdm
66
transformer_engine[pytorch]
7-
transformers
7+
transformers<5.0
88
wandb

bionemo-recipes/recipes/geneformer_native_te_mfsdp_fp8/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,5 +6,5 @@ torch
66
torchao!=0.14.0
77
tqdm
88
transformer_engine
9-
transformers
9+
transformers<5.0
1010
wandb

bionemo-recipes/recipes/llama3_native_te/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ torchdata
66
torchmetrics
77
tqdm
88
transformer_engine[pytorch]
9-
transformers
9+
transformers<5.0
1010
wandb
1111
zstandard
1212
nvdlfw_inspect @ git+https://github.com/NVIDIA/nvidia-dlfw-inspect

0 commit comments

Comments
 (0)