colt5-attention>=0.8.1
einops>=0.6.1
local-attention>=1.8.6
torch>=1.6
