[project]
name = "progressive-llm-training"
version = "0.1.0"
description = "Progressive LLM Training for 松尾研LLMコンペ2025"
requires-python = ">=3.9"

dependencies = [
    "torch>=2.0.0",
    "transformers>=4.40.0",
    "accelerate>=0.27.0",
    "peft>=0.11.0",
    "trl>=0.9.0",
    "datasets>=2.18.0",
    "bitsandbytes>=0.43.0",
    "wandb>=0.16.0",
    "pyyaml>=6.0",
    "jsonlines>=4.0.0",
    "deepspeed>=0.12.0",
]

[project.optional-dependencies]
dev = ["pytest", "black", "isort"]

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"