pydantic<2.7
torch>=1.7.0
transformers<4.41

[dev]
black==22.12.0
flake8>=3.8.3
isort==5.8.0
nbconvert>=7.16.3
pytest>=6.0.0
wheel>=0.36.2
