VAE encoding images uses about 5G of memory. Very High!!

When I execute the code locally, the memory usage is very high when VAE encodes images. Do I need to initialize some parameters? If necessary, which parameters should I configure?

When running in the deployment environment, VAE encoding images uses about 1G of memory. However, when I execute the code directly locally, the memory usage is about 5G.