mirror of
https://github.com/wassname/cookiecutter-data-science.git
synced 2026-07-20 12:20:46 +08:00
Update pyproject.toml
This commit is contained in:
@@ -7,25 +7,20 @@ license = "MIT"
|
||||
readme = "README.md"
|
||||
requires-python = ">=3.10"
|
||||
dependencies = [
|
||||
python = ">=3.10"
|
||||
numpy = "^1.26.1"
|
||||
pandas = "^2.1.1"
|
||||
matplotlib = "^3.8.0"
|
||||
scikit-learn = "^1.3.1"
|
||||
loguru = "^0.7.2"
|
||||
tqdm = "^4.66.1"
|
||||
# einops = "^0.7.0"
|
||||
# simple-parsing = "^0.1.4"
|
||||
# torch = {version = "^2.1.0+cu124", source = "pytorch"}
|
||||
# lightning = "^2.1.0"
|
||||
# pytorch-optimizer = "^2.12.0"
|
||||
# torchinfo = "^1.8.0"
|
||||
# accelerate = "^0.24.1"
|
||||
# transformers = "4.34.0"
|
||||
# accelerate = "^0.24.1"
|
||||
# peft = "^0.5.0"
|
||||
# loguru = "^0.7.2"
|
||||
# einops = "^0.3.1"
|
||||
"numpy=^1.26.1",
|
||||
"pandas=^2.1.1",
|
||||
"matplotlib=^3.8.0",
|
||||
"scikit-learn=^1.3.1",
|
||||
"loguru=^0.7.2",
|
||||
"tqdm=^4.66.1",
|
||||
"transformers[torch]=^4.34.0",
|
||||
# "einops=^0.7.0",
|
||||
# "loguru=^0.7.2",
|
||||
# "simple-parsing=^0.1.4",
|
||||
# "torch =^2.1.0+cu124",
|
||||
# "lightning=^2.1.0",
|
||||
# "torchinfo=^1.8.0",
|
||||
# "peft=^0.5.0",
|
||||
]
|
||||
|
||||
[dependency-groups]
|
||||
|
||||
Reference in New Issue
Block a user