From a90a96376516e968557730456a0e8e83d992ebaf Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Thu, 23 Nov 2023 23:22:05 -0600
Subject: [PATCH] fix(api): ensure pipeline is using selected scheduler

---
 api/onnx_web/diffusers/load.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py
index 75dafc2e..a04cb5e5 100644
--- a/api/onnx_web/diffusers/load.py
+++ b/api/onnx_web/diffusers/load.py
@@ -227,6 +227,9 @@ def load_pipeline(
             **components,
         )
 
+        if pipe.scheduler != scheduler:
+            pipe.scheduler = scheduler
+
         # make sure XL models are actually being used
         if "text_encoder_session" in components:
             pipe.text_encoder = ORTModelTextEncoder(