NotImplementedError：找不到 Memory_efficient_attention_forward 的运算符

Question

在 google colab 中运行 stable diffsion notebook 时出现以下错误。

NotImplementedError: No operator found for `memory_efficient_attention_forward` with inputs:
 query       : shape=(1, 4096, 1, 512) (torch.float16)
 key         : shape=(1, 4096, 1, 512) (torch.float16)
 value       : shape=(1, 4096, 1, 512) (torch.float16

更新了 xformers 但不起作用

Answer 1

!pip install --pre -U xformers

参考 这解决了我的问题。