einops>=0.6.1
torch>=1.6
colt5-attention>=0.6.2
