Try out the pre-release like this:

`pip install imaginairy==6.0.0a0 --upgrade`

New 512x512 model supported with all samplers and inpainting

New 768x768 model supported with the DDIM sampler only

Not yet supported is the upscaling and depth maps.

To be honest I'm not sure the new model produces better images but maybe they will release some improved models in the future now that they have the pipeline open.