How do you handle the info loss from VAE encoding?

[Darkbblue]

Using internal features from a LATENT diffusion model means the width / height of what you get is only 1/8 of the original image. I'm trying to implement my own feature extraction pipeline and have encountered this info loss issue, which makes my segmentation results a total mess. I have also read the codes in this repo but not figured out how you manage to handle it. That's why I'm here asking. Thanks in advance!