torch<3.0.0,>=2.3.0
transformers<5.0.0,>=4.38.0
peft<1.0.0,>=0.9.0
datasets<5.0.0,>=2.21.0
accelerate<2.0.0,>=0.27.0
pyyaml<7.0.0,>=6.0.1
pydantic<3.0.0,>=2.0.0
trl<2.0.0,>=0.12.0
tensorboard<3.0.0,>=2.15.0
huggingface_hub<1.0.0,>=0.23.2
requests<3.0.0,>=2.31.0

[chat]
rich>=13.0.0

[dev]
pytest>=8.0.0
pytest-cov>=5.0.0
ruff>=0.4.0

[distributed]
deepspeed>=0.14.0

[eval]
lm-eval<1.0.0,>=0.4.4

[export]

[export:sys_platform != "win32"]
llama-cpp-python>=0.2.90

[ingestion]
pypdf<6.0.0,>=4.0.0
python-docx<2.0.0,>=1.0.0
ebooklib<1.0.0,>=0.18
beautifulsoup4<5.0.0,>=4.12.0
langdetect<2.0.0,>=1.0.9
xxhash<4.0.0,>=3.0.0

[ingestion-pii-ml]
presidio-analyzer<3.0.0,>=2.2.0

[ingestion-scale]
datasketch<2.0.0,>=1.6.0

[merging]
mergekit>=0.0.5

[qlora]

[qlora:sys_platform == "linux"]
bitsandbytes<1.0.0,>=0.42.0

[security]
pip-audit<3.0.0,>=2.7.0
bandit[toml]<2.0.0,>=1.7.0

[tracking]
wandb>=0.16.0

[unsloth]

[unsloth:sys_platform == "linux"]
unsloth>=2024.3.0
