Update app.py

app.py

@@ -19,8 +19,9 @@ 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)
+        if model_id == "None1145/noobai-XL-Vpred-0.65s-openvino" or "Vpred" in model_id or "vpred" in model_id or "v-pred" in model_id:
+            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()