Support for higher non-squared image resolutions ?

[AshAJ1994]

Hi,

Thanks for the great work.

I would like to check if it could also work for other datasets with higher non-squared resolutions beyond 256x256 ? Also, can it be combined with conditions coming from text prompts as well ?

Thank you.

Regards,
Ashish