Update app.py

app.py

@@ -25,7 +25,7 @@ class CustomOVModelVaeDecoder(OVModelVaeDecoder):
         super(OVModelVaeDecoder, self).__init__(model, parent_model, ov_config, "vae_decoder", model_dir)
 
 
-pipe = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_config = {"CACHE_DIR":""}, )
+pipe = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_config = {"CACHE_DIR":""}, torch_dtype=torch.int8, )
 
 taesd_dir = snapshot_download(repo_id="deinferno/taesd-openvino")
 pipe.vae_decoder = CustomOVModelVaeDecoder(model = OVBaseModel.load_model(f"{taesd_dir}/vae_decoder/openvino_model.xml"), parent_model = pipe, model_dir = taesd_dir)