rich>=13.0
click>=8.1
pydantic>=2.0
httpx>=0.25
fastapi>=0.110.0
uvicorn>=0.29.0
websockets>=12.0

[all]
shell-whisperer[export,gguf,gpu,train]

[dev]
pytest>=8.0.0
pytest-asyncio>=0.23.0
ruff>=0.3.0
mypy>=1.9.0

[export]
onnx>=1.16.0
onnxruntime>=1.17.0
optimum>=1.17.0

[gguf]
llama-cpp-python>=0.2.50

[gpu]
bitsandbytes>=0.43.0
xformers>=0.0.23

[train]
torch>=2.1.0
transformers>=4.40.0
peft>=0.10.0
datasets>=2.18.0
unsloth>=2024.1
accelerate>=0.27.0
trl>=0.8.0
wandb>=0.16.0
