There is a bug in FourierBlock

[ZhijiangTang]

Thanks for your surprising work!
There is a bug in FourierBlock. It should be n_heads, but it is an exact number of 8. The code would usually work if n_heads=8, but it would throw an error if n_heads is not equal to 8.