It previously only supported 512x512 images. Now, it supports other dimensions. You can reconvert the model with a specific image dimension by following the steps noted in this comment.