Inference with a 512×512 size picture resulted in the following error where I changed the latent_size = 64 because I couldn't reason with a 512×512 size picture when latent_size=32
What should I do because I need to reconstruct the different sizes picture in the first stage