Updated it to PyTorch2.0 and replaced Attention with PyTorch's Multihead attention

[Mut1nyJD]

I've upgraded the code to be compatible with PyTorch2.0 and also replaced the attention/crossattention node to use PyTorch's build-in Multihead Attention which also of course supports flash attention out of the box.

Is there any interest in these updates?

[VeinVevive]

I am interested in it, could you send me the code?Thank you！