Update app.py

app.py

@@ -15,7 +15,7 @@ model_id = "Disty0/LCM_SoteMix"
 
 #1024*512 記憶體不足
 HIGH=768
-WIDTH=768
+WIDTH=512
 
 batch_size = -1
 class CustomOVModelVaeDecoder(OVModelVaeDecoder):
@@ -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":""}, torch_dtype=torch.bool, )
+pipe = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_config = {"CACHE_DIR":""}, torch_dtype=torch.bfloat16, )
 
 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)