torch>=2.1.0
transformers>=4.40.0
numpy
huggingface-hub
pydantic>=2.0.0

[serve]
fastapi
uvicorn

[test]
pytest
