Update app.py

app.py

@@ -16,6 +16,21 @@ model_id = "OpenVINO/LCM_Dreamshaper_v7-int8-ov"
 pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False)
 
 batch_size, num_images, height, width = 1, 1, 1024, 512
+
+
+class CustomOVModelVaeDecoder(OVModelVaeDecoder):
+    def __init__(
+        self, model: ov.Model, parent_model: OVBaseModel, ov_config: Optional[Dict[str, str]] = None, model_dir: str = None,
+    ):
+        super(OVModelVaeDecoder, self).__init__(model, parent_model, ov_config, "vae_decoder", model_dir)
+
+
+pipeline = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_config = {"CACHE_DIR":""})
+
+taesd_dir = snapshot_download(repo_id="deinferno/taesd-openvino")
+pipeline.vae_decoder = CustomOVModelVaeDecoder(model = OVBaseModel.load_model(f"{taesd_dir}/vae_decoder/openvino_model.xml"), parent_model = pipeline, model_dir = taesd_dir)
+
+
 pipeline.reshape(batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images)
 
 #不可用lora