[Bug] MemoryEfficientAttentionTritonFwdFlashBwOp no longer availale

[imD-5]

Hi! thanks for this great repo it's helping me a lot.
So i've upgraded to xformers == 0.0.25, stable-fast==1.0.4, and i get the following error.

AttributeError: module 'xformers.ops' has no attribute 'MemoryEfficientAttentionTritonFwdFlashBwOp'. Did you mean: 'MemoryEfficientAttentionCutlassFwdFlashBwOp'?

I did some research and apparently xformers deleted the function MemoryEfficientAttentionTritonFwdFlashBwOp because it wasn't as efficient and also hgave some errors under some circumstances.

reference: https://github.com/facebookresearch/xformers/blob/main/CHANGELOG.md

can you fix this?

[jryebread]

same issue :(