apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): ensure pipeline is using selected scheduler

This commit is contained in:
Sean Sube 2023-11-23 23:22:05 -06:00
parent ddd3d45b79
commit a90a963765
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/onnx_web/diffusers/load.py
View File

@ -227,6 +227,9 @@ def load_pipeline(
**components, **components,
) )
if pipe.scheduler != scheduler:
pipe.scheduler = scheduler
# make sure XL models are actually being used # make sure XL models are actually being used
if "text_encoder_session" in components: if "text_encoder_session" in components:
pipe.text_encoder = ORTModelTextEncoder( pipe.text_encoder = ORTModelTextEncoder(