No module named localAttention

[feihuzhang]

Thanks for your nice work.
It seems the localAttention lib is missing.

SwissArmyTransformer/SwissArmyTransformer/ops/local_attention_function.py

Line 5 in 85d0182

 from localAttention import (similar_forward, 

Did you use this? https://github.com/zzd1992/Image-Local-Attention
But, I see there are some differences.

Thanks.

[Sleepy_chord]

please use

git clone https://github.com/Sleepychord/Image-Local-Attention
cd Image-Local-Attention && python setup.py install

I forked the Zhendong's repo and change the implementation for speed up and more functions.