23 lines
516 B
TOML
23 lines
516 B
TOML
[build-system]
|
|
requires = ["setuptools>=68", "wheel"]
|
|
build-backend = "setuptools.build_meta"
|
|
|
|
[project]
|
|
name = "verysimplemoe"
|
|
version = "0.1.0"
|
|
description = "A very small PyTorch MoE language model trainer for FineWeb"
|
|
requires-python = ">=3.10"
|
|
dependencies = [
|
|
"torch>=2.2",
|
|
"transformers>=4.40",
|
|
"datasets>=2.19",
|
|
"tqdm>=4.66",
|
|
]
|
|
|
|
[project.scripts]
|
|
verysimplemoe-train = "verysimplemoe.train:main"
|
|
verysimplemoe-generate = "verysimplemoe.generate:main"
|
|
|
|
[tool.setuptools.packages.find]
|
|
where = ["src"]
|