Skip to content

Commit 4264120

Browse files
authored
Pin transformers (#998)
Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>
1 parent 4d79815 commit 4264120

File tree

2 files changed

+463
-163
lines changed

2 files changed

+463
-163
lines changed

pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ dependencies = [
5252
"pyarrow",
5353
"ray[default,data]>=2.49",
5454
"torch",
55-
"transformers>=4.48.0",
55+
"transformers==4.53.1",
5656
]
5757

5858
[project.optional-dependencies]
@@ -133,7 +133,7 @@ internvideo2 = [
133133
"torchaudio>=2.4.1",
134134
"torchvision>=0.19.1",
135135
"tqdm>=4.66.5",
136-
"transformers>=4.45.1",
136+
"transformers==4.53.1",
137137
"wandb>=0.18.3",
138138
]
139139

0 commit comments

Comments
 (0)