torch
transformers>4.45.2
accelerate

[test]
pytest
