[project]
name = "sec-cybert-train"
version = "0.1.0"
description = "SEC-cyBERT training pipeline: DAPT, TAPT, fine-tuning, and evaluation"
readme = "README.md"
requires-python = ">=3.13"
dependencies = [
    "torch",
    "transformers",
    "datasets",
    "accelerate",
    "pyyaml",
]

[project.optional-dependencies]
flash = ["flash-attn"]
decoder = ["unsloth"]

[project.scripts]
sec-cybert = "main:main"