if supports outputting the edited image according to the size of the original image?

[lizekui]

The project is great! I wonder if it supports outputting the edited image according to the size of the original image?

I try some result but I found the output on replicate and hugginface version is square picture

Thanks firstly!