update vae

app.py

@@ -73,7 +73,7 @@ net.load_state_dict(torch.load(path_model, weights_only=False))
 net.eval().to(device)
 
 # Initialize VAE (only Decoder will be used)
-vae = AutoencoderKL.from_pretrained("CompVis/stable-diffusion-v1-4", subfolder="vae").eval().to(device)
+vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").eval().to(device)
 
 # Initialize the upscaler
 upscaler = UpscalerESRGAN(