Skip to content

Commit 98e01c9

Browse files
feat: version freeze
Added versions to freeze issues
1 parent 025a7ac commit 98e01c9

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

pyproject.toml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -44,11 +44,11 @@ dependencies = [
4444
"torch >= 1.9.0",
4545
"pytorch-lightning",
4646
"transformers==4.31.0",
47-
"datasets",
48-
"evaluate",
47+
"datasets==2.14.5",
48+
"evaluate==0.4.0",
4949
"bitsandbytes==0.41.1",
5050
"sentencepiece",
51-
"deepspeed",
51+
"deepspeed==0.9.5",
5252
"gradio",
5353
"click",
5454
"wget",
@@ -58,7 +58,7 @@ dependencies = [
5858
"openai >= 0.27.0",
5959
"pydantic >= 1.10.0",
6060
"rouge-score >= 0.1.2",
61-
"accelerate",
61+
"accelerate==0.22.0",
6262
"wandb",
6363
]
6464

0 commit comments

Comments
 (0)