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.1.1
instructlab-quantize>=0.1.0
instructlab-schema>=0.3.1
instructlab-sdg>=0.2.4
instructlab-training>=0.3.1
jsonschema>=4.21.1
llama_cpp_python[server]==0.2.79
openai>=1.13.3
peft>=0.9.0
platformdirs>=4.2
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
sentencepiece>=0.2.0
tokenizers>=0.11.1
toml>=0.10.2
tqdm>=4.66.2
trl>=0.9.4
wandb>=0.16.4
yamllint>=1.35.1
referencing>=0.35.0

[:python_version == "3.10"]
numpy<2.0.0,>=1.23.5
torch<2.4.0,>=2.2.2a0
transformers>=4.40.0

[:python_version >= "3.11"]
numpy<2.0.0,>=1.26.4
torch<2.4.0,>=2.3.0
transformers>=4.41.2

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

[cpu]

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

[hpu]
optimum>=1.20.1
optimum-habana>=1.12.0
numpy<2.0.0,>=1.23.5
torch<2.4.0,>=2.2.2a0
habana-torch-plugin>=1.16.2
habana_gpu_migration>=1.16.2

[mps]

[rocm]
