click<9.0.0,>=8.1.7
click-didyoumean>=0.3.0
datasets>=2.18.0
gguf>=0.6.0
GitPython>=3.1.42
httpx>=0.25.0
instructlab-eval>=0.3.1
instructlab-quantize>=0.1.0
instructlab-schema>=0.4.0
instructlab-sdg>=0.3.0
instructlab-training>=0.5.0
llama_cpp_python[server]==0.2.79
numpy<2.0.0,>=1.26.4
openai>=1.13.3
peft>=0.9.0
prompt-toolkit>=3.0.38
pydantic>=2.7.4
pydantic_yaml>=1.2.0
PyYAML>=6.0.0
rich>=13.3.1
rouge-score>=0.1.2
ruamel.yaml>=0.17.0
sentencepiece>=0.2.0
tokenizers>=0.11.1
toml>=0.10.2
torch<2.5.0,>=2.3.0
tqdm>=4.66.2
transformers>=4.41.2
trl>=0.9.4
wandb>=0.16.4
xdg-base-dirs>=6.0.1
psutil>=6.0.0

[:sys_platform == "darwin" and platform_machine == "arm64"]
mlx<0.6.0,>=0.5.1

[cpu]

[cuda]
instructlab-training[cuda]>=0.5.0

[hpu]
optimum>=1.21.0
optimum-habana>=1.13.1
torch<2.4.0,>=2.3.1a0
habana-torch-plugin>=1.17.1
habana_gpu_migration>=1.17.1

[mps]

[rocm]
instructlab-training[rocm]>=0.5.0
