torch
transformers
pydantic==1.*