26 lines
579 B
TOML
26 lines
579 B
TOML
[build-system]
|
|
requires = ["setuptools>=42", "wheel"]
|
|
build-backend = "setuptools.build_meta"
|
|
|
|
[project]
|
|
name = "gemma_train"
|
|
version = "0.1.0"
|
|
requires-python = ">=3.8"
|
|
|
|
dependencies = [
|
|
"accelerate==1.2.0",
|
|
"bitsandbytes>=0.45.0",
|
|
"datasets==3.3.2",
|
|
"evaluate==0.4.3",
|
|
"kernels>=0.11.1",
|
|
"llm_engine_utils @ file:///root/sia/lib/llm_engine_utils",
|
|
"peft==0.13.2",
|
|
"python-dotenv>=1.0.0",
|
|
"sentencepiece>=0.2.0",
|
|
"torch==2.4.1",
|
|
"transformers==4.51.3",
|
|
"trl>=0.17.0",
|
|
]
|
|
|
|
[project.scripts]
|
|
gemma_train = "gemma_train.__main__:main" |