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