fix(api): ensure pipeline is using selected scheduler
This commit is contained in:
parent
ddd3d45b79
commit
a90a963765
|
@ -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(
|
||||
|
|
Loading…
Reference in New Issue