i want to change generation size from 512 x 512 to 768x768 which one should i change on latent_diffusion.py??