NotImplementedError: No operator found for memory_efficient_attention_forward with inputs: query

[seoyeon76]

NotImplementedError: No operator found for memory_efficient_attention_forward with inputs: query : shape=(2, 12500, 8, 40) (torch.float16) key : shape=(2, 12500, 8, 40) (torch.float16) value : shape=(2, 12500, 8, 40) (torch.float16) attn_bias : <class 'NoneType'> p : 0.0 cutlassF is not supported because: xFormers wasn't build with CUDA support flshattF is not supported because: xFormers wasn't build with CUDA support tritonflashattF is not supported because: xFormers wasn't build with CUDA support requires A100 GPU smallkF is not supported because: xFormers wasn't build with CUDA support dtype=torch.float16 (supported: {torch.float32}) max(query.shape[-1] != value.shape[-1]) > 32 unsupported embed per head: 40

easy-stable-diffusion.ipynb - Colab - Google

I'm working on it here, but it doesn't work.
I'm a beginner.
Please help me

[cperry-goog]

You may have more luck contacting the notebook author if you're having a hard time with code. If you share a minimal reproducible notebook we can investigate to see if it's an issue with Colab.