cehrbert==1.3.3
openai==1.54.3
optuna==4.0.0
transformers==4.40.0
tokenizers==0.19
trl==0.11.4

[dev]
pre-commit
pytest
pytest-cov
pytest-subtests
rootutils
hypothesis
black

[flash_attn]
flash_attn
