OpenPeer-AI-NTK-Trainer / pyproject.toml
Mentors4EDU's picture
Upload 41 files
3f2dde4 verified
[build-system]
requires = ["setuptools>=68", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "openpeer-ntk-trainer"
version = "0.1.0"
description = "OpenPeerLLM trainer using ntkmirror controllers and a tinygrad gate demo"
readme = "README.md"
requires-python = ">=3.10"
dependencies = [
"torch>=2.2",
"transformers>=4.42",
"pandas>=2.2",
"plotly>=6.0",
"psutil>=6.0",
]
[project.optional-dependencies]
demo = ["tinygrad>=0.10.0"]
charts = ["openbb>=4.0"]
gui = ["streamlit>=1.36"]
ntk = ["ntkmirror @ git+https://github.com/leochlon/ntkmirror.git"]
all = ["tinygrad>=0.10.0", "openbb>=4.0", "streamlit>=1.36", "ntkmirror @ git+https://github.com/leochlon/ntkmirror.git"]
[project.scripts]
openpeer-trainer = "openpeer_trainer.cli:main"
[tool.setuptools]
package-dir = {"" = "src"}
[tool.setuptools.packages.find]
where = ["src"]