OchiaiNao

KeyError in visualize attn maps

4

I am using the notebook given #54 When I run the loading part: ```python ldm_stable.load_model('/content/drive/MyDrive/diffusion/logs/.bin') ``` Returns this: ``` --------------------------------------------------------------------------- KeyError Traceback (most recent call last) [](https://localhost:8080/#) in () 9...