gradio<7,>=6.16.0
transformers<6,>=5.10.1
datasets<6,>=5.0.1
peft<1,>=0.18.0
accelerate<2,>=1.15.0
bitsandbytes>=0.45.0
trl<1.15,>=0.29.1
torch<3,>=2.13.0
numpy>=2.0.0
pandas>=2.2.0
safetensors>=0.4.3
tqdm>=4.66.3
einops>=0.8.0
matplotlib>=3.9.0
huggingface_hub<3,>=1.5.0
typer>=0.15.0
pypdf>=6.19.0
openpyxl>=3.1.0
python-docx>=1.1.0
psutil>=6.0.0

[all]
llm-finetuner-pro[dev,embedding,eval,heretic,vision,vllm]

[compress]
llmcompressor>=0.14.0

[dev]
black>=26.3.1
pytest>=9.0.3
ruff>=0.4.0

[embedding]
sentence-transformers<7,>=5.6.0

[eval]
evaluate>=0.4.0
rouge-score>=0.1.2
bert-score>=0.3.13
nlpaug>=1.1.10
lm-eval<0.5,>=0.4.13
math-verify>=0.5.2

[heretic]
heretic-llm>=1.2.0

[vision]
torchvision

[vllm]
trl[vllm]<1.15,>=0.29.1
