[Tests] Test attention.py (#368)
[Tests] Test attention.py (#368)

* add test for AttentionBlock, SpatialTransformer

* add context_dim, handle device

* removed dropout test

* fixes, add dropout test
huggingface/diffusers