26 lines
No EOL
574 B
TOML
26 lines
No EOL
574 B
TOML
[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" |