transformers
torch

[dev]
pytest
black
ruff

[test]
pytest
