悪いことは言わんから横幅は536か544にしたほうがええで
SDは全て1/8のエンコードと8倍のデコードしてるから8の倍数以外のサイズ指定は端数をエンコーダーが認識出来なくて破綻の元や
ついでに言うとpytorchやcudaの処理は64の倍数が都合いいから画像サイズの縦横は64の倍数推奨なんやで