loaded vae

app.py

@@ -14,8 +14,10 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
 # Load the initial VAE model for preprocessing YAY
-vae_model_name = "runwayml/stable-diffusion-v1-5"  # Changed VAE model
-vae = AutoencoderKL.from_pretrained(vae_model_name).to(device)
+# Load the initial VAE model for preprocessing
+vae_model_name = "runwayml/stable-diffusion-v1-5"  # Adjusted VAE model path
+vae = AutoencoderKL.from_pretrained(vae_model_name, subfolder="vae").to(device)
+
 
 # Load the FLUX diffusion pipeline with optimizations
 pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype)