We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 025a7ac commit 98e01c9Copy full SHA for 98e01c9
pyproject.toml
@@ -44,11 +44,11 @@ dependencies = [
44
"torch >= 1.9.0",
45
"pytorch-lightning",
46
"transformers==4.31.0",
47
- "datasets",
48
- "evaluate",
+ "datasets==2.14.5",
+ "evaluate==0.4.0",
49
"bitsandbytes==0.41.1",
50
"sentencepiece",
51
- "deepspeed",
+ "deepspeed==0.9.5",
52
"gradio",
53
"click",
54
"wget",
@@ -58,7 +58,7 @@ dependencies = [
58
"openai >= 0.27.0",
59
"pydantic >= 1.10.0",
60
"rouge-score >= 0.1.2",
61
- "accelerate",
+ "accelerate==0.22.0",
62
"wandb",
63
]
64
0 commit comments