21 lines
410 B
TOML
21 lines
410 B
TOML
[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"
|