Update handler.py

handler.py

@@ -8,14 +8,6 @@ from diffusers.image_processor import VaeImageProcessor
 class EndpointHandler:
     def __init__(self, path=""):
         kwargs = {"torch_dtype": torch.float16}
-        model_data = model_info(path)
-        has_model_index = any(
-            file.rfilename == "model_index.json" for file in model_data.siblings
-        )
-
-        if has_model_index:
-            kwargs["subfolder"] = "vae"
-
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         self.dtype = kwargs["torch_dtype"]
         self.vae = AutoencoderKL.from_pretrained(path, **kwargs).to(self.device).eval()