Update app.py

app.py

@@ -4,12 +4,13 @@ from optimum.intel.openvino.modeling_diffusion import OVModelVaeDecoder, OVBaseM
 from huggingface_hub import snapshot_download
 import openvino.runtime as ov
 from typing import Optional, Dict
+from diffusers import DiffusionPipeline, EulerDiscreteScheduler, KDPM2DiscreteScheduler
 
 model_id = "hsuwill000/AngelBreed-v4-openvino-8bit"
 
 #只有512x512 否則一直重畫
 HIGH = 512
-WIDTH = 512
+WIDTH = 1024
 
 batch_size = -1  # Or set it to a specific positive integer if needed
 """
@@ -29,6 +30,8 @@ examples = [
     "((colofrul:1.7)),((best quality)), ((masterpiece)), ((ultra-detailed)), (illustration), (detailed light), (an extremely delicate and beautiful),incredibly_absurdres,(glowing),(1girl:1.7),solo,a beautiful girl,(((cowboy shot))),standding,((Hosiery)),((beautiful off-shoulder lace-trimmed layered strapless dress+white stocking):1.25),((Belts)),(leg loops),((Hosiery)),((flower headdress)),((long white hair)),(((beautiful eyes))),BREAK,((english text)),(flower:1.35),(garden),(((border:1.75))),",
 ]
 
+scheduler = KDPM2DiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
+                     
 pipe = OVStableDiffusionPipeline.from_pretrained(
     model_id,
     compile=False,
@@ -59,6 +62,7 @@ negative_prompt = "Easy Negative, worst quality, low quality, normal quality, lo
 def infer(prompt, negative_prompt):
     image = pipe(
         prompt=prompt,
+        scheduler=scheduler,
         negative_prompt=negative_prompt,
         width=WIDTH,
         height=HIGH,