transformers>=4.37.0
accelerate
torch
triton
flash_attn
pycuda==2023.1

[dev]
transformers>=4.37.0
accelerate
torch
triton
flash_attn
pycuda==2023.1
black==21.4b0
flake8>=3.8.3
isort>=5.5.4
pre-commit
pytest
pytest-xdist

[quality]
black==21.4b0
flake8>=3.8.3
isort>=5.5.4
pre-commit
pytest
pytest-xdist
