[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"