requirements-dev.txt

# formatting
yapf==0.32.0
toml==0.10.2
ruff==0.1.5

# type checking
mypy==0.991
types-PyYAML
types-requests
types-setuptools

# testing
pytest
pytest-forked
pytest-asyncio
httpx
einops # required for MPT
flash_attn # required for HuggingFace's llama implementation
openai
requests
ray