torch>=2.0.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

[vllm]
vllm<0.20.0,>=0.19.1

[vllm20]
vllm<0.21.0,>=0.20.0
