Update app.py

app.py

@@ -3,6 +3,7 @@ import numpy as np
 import random
 from optimum.intel import OVStableDiffusionXLPipeline
 import torch
+from diffusers import EulerDiscreteScheduler
 
 model_id = "None1145/noobai-XL-Vpred-0.65s-openvino"
 
@@ -18,6 +19,8 @@ def reload_model(new_model_id):
     try:
         print(f"{model_id}...")
         pipe = OVStableDiffusionXLPipeline.from_pretrained(model_id, compile=False)
+        scheduler_args = {"prediction_type": "v_prediction", "rescale_betas_zero_snr": True}
+        pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, **scheduler_args)
         # pipe.to("gpu")
         pipe.reshape(batch_size=1, height=prev_height, width=prev_width, num_images_per_prompt=1)
         pipe.compile()