New 512x with image inputs

Hi there, I just wanted to say this is fantastic work you've done here, I'm amazed at the results and really look forward to how your team progresses with this.

As for my query, I've been trying to work out how to use image prompts for the new 512x512 notebook (Malevich-3.5GB-vRAM-usage.ipynb) but I haven't managed to make it work, I was wondering if you could help?