torch>=2.10.0
transformers>=5.5.1

[build]
huggingface_hub
pyyaml
tqdm
safetensors

[compose]
huggingface_hub
pyyaml
tqdm
safetensors

[dev]
pytest
granite-switch[compose,hf,vllm]

[dev-vllm20]
pytest
granite-switch[compose,hf,vllm20]

[hf]
accelerate>=0.20.0

[test]
pytest
bitsandbytes
optimum-quanto

[tutorials]
granite-switch[compose,hf,vllm]
chromadb>=0.4.0
httpx>=0.24.0
requests>=2.31.0
rich>=13.0.0
mellea>=0.1.0
ipython>=8.10.0
python-dotenv>=1.0.0

[vllm]
vllm<0.20.0,>=0.19.1

[vllm20]
vllm<0.21.0,>=0.20.0
