Skip to content

Commit fc8721b

Browse files
authored
Update pyproject.toml
1 parent c858786 commit fc8721b

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

pyproject.toml

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,12 +21,11 @@ dependencies = [
2121

2222
# This should be main eval related (eg. flashattention?)
2323
[project.optional-dependencies]
24-
eval = ["datasets", "ipykernel", "packaging", "flash-attn==2.6.1", "pandas"]
24+
eval = ["protobuf","datasets", "packaging", "flash-attn==2.6.1", "pandas"]
2525

26-
# torch==2.4.0, transformers==4.44.1, accelerate, sentencepiece, protobuf, datasets, ipykernel, packaging, flash-attn==2.6.1, pandas
2726

2827
[project.urls]
2928
"Homepage" = "TODO"
3029

3130
[tool.setuptools.packages.find]
32-
exclude = ["notebooks*", "scripts*"]
31+
exclude = ["notebooks*", "scripts*"]

0 commit comments

Comments
 (0)