Update app.py

app.py

@@ -28,14 +28,18 @@ class CustomOVModelVaeDecoder(OVModelVaeDecoder):
 pipe = OVStableDiffusionPipeline.from_pretrained(
         model_id, 
         compile = False, 
-        ov_config = {"CACHE_DIR":""}, 
-        torch_dtype=torch.bfloat16, 
-        variant="fp16",
+        ov_config = {"CACHE_DIR":""},
+        torch_dtype=torch.int8,
+        #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)
+pipe.vae_decoder = CustomOVModelVaeDecoder(model = OVBaseModel.load_model(f"{taesd_dir}/vae_decoder/openvino_model.xml"), 
+                                           parent_model = pipe, 
+                                           model_dir = taesd_dir
+                                          )
 pipe.reshape( batch_size=-1, height=HIGH, width=WIDTH, num_images_per_prompt=1)
 #pipe.load_textual_inversion("./badhandv4.pt", "badhandv4")
 #pipe.load_textual_inversion("./Konpeto.pt", "Konpeto")