Update app.py

app.py

@@ -25,7 +25,14 @@ 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":""}, torch_dtype=torch.bfloat16, )
+pipe = OVStableDiffusionPipeline.from_pretrained(
+        model_id, 
+        compile = False, 
+        ov_config = {"CACHE_DIR":""}, 
+        torch_dtype=torch.bfloat16, 
+        variant="fp16",
+        use_safetensors=False,
+        )
 
 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)