Update app.py

app.py

@@ -13,25 +13,27 @@ model_id = "Disty0/SoteMixV3"
 #model_id = "Disty0/sotediffusion-v2" #不可
 
 #1024*512 記憶體不足 1024x1536 
-HIGH=1024
-WIDTH=1536
+HIGH=512
+WIDTH=512
 
 
 batch_size = -1
-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)
+#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)
 
 
-pipe = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_config = {"CACHE_DIR":""})
+pipe = OVStableDiffusionPipeline.from_pretrained(model_id)
+#pipe = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_config = {"CACHE_DIR":""})
 
 #有taesd很醜
 #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.reshape( batch_size=-1, height=HIGH, width=WIDTH, num_images_per_prompt=1)
+#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")
 #<shigure-ui-style>
@@ -40,7 +42,7 @@ pipe.reshape( batch_size=-1, height=HIGH, width=WIDTH, num_images_per_prompt=1)
 #pipe.load_textual_inversion("sd-concepts-library/agm-style-nao")
 
 
-pipe.compile()
+#pipe.compile()
 
 prompt=""
 negative_prompt="(worst quality, low quality, lowres), zombie, interlocked fingers,"
@@ -80,7 +82,7 @@ with gr.Blocks(css=css) as demo:
     
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""
-        # Disty0/SoteMixV3 {HIGH}x{WIDTH}
+        # 不符合預期圖片, Disty0/SoteMixV3 {HIGH}x{WIDTH}
         Currently running on {power_device}.
         """)