changing sd-vae-ft-mse path

handler.py

@@ -39,7 +39,13 @@ class EndpointHandler():
         self._initialize_pipeline()
 
     def _initialize_pipeline(self):
-        vae = AutoencoderKL.from_pretrained('./pretrained_weights/sd-vae-ft-mse').to(device, dtype=self.weight_dtype)
+        base_dir = os.path.dirname(os.path.abspath(__file__))
+        config_path = os.path.join(base_dir, 'pretrained_weights', 'sd-vae-ft-mse')
+
+        if not os.path.exists(config_path):
+            raise FileNotFoundError(f"The sd-vae-ft-mse folder was not found at: {config_path}")
+
+        vae = AutoencoderKL.from_pretrained(config_path).to(device, dtype=self.weight_dtype)
 
         reference_unet = UNet2DConditionModel.from_pretrained(
             self.config.pretrained_base_model_path,