1
0
Fork 0

avoid passing None as encoder

This commit is contained in:
Sean Sube 2023-02-25 13:12:58 -06:00
parent 3626d69f40
commit 18f59f034d
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 10 additions and 10 deletions

View File

@ -179,17 +179,18 @@ def load_pipeline(
custom_pipeline = None
logger.debug("loading new diffusion pipeline from %s", model)
scheduler = scheduler_type.from_pretrained(
model,
provider=device.ort_provider(),
sess_options=device.sess_options(),
subfolder="scheduler",
)
components = {
"scheduler": scheduler_type.from_pretrained(
model,
provider=device.ort_provider(),
sess_options=device.sess_options(),
subfolder="scheduler",
)
}
text_encoder = None
if inversion is not None:
logger.debug("loading text encoder from %s", inversion)
text_encoder = OnnxRuntimeModel.from_pretrained(
components["text_encoder"] = OnnxRuntimeModel.from_pretrained(
path.join(inversion, "text_encoder"),
provider=device.ort_provider(),
sess_options=device.sess_options(),
@ -202,8 +203,7 @@ def load_pipeline(
sess_options=device.sess_options(),
revision="onnx",
safety_checker=None,
scheduler=scheduler,
text_encoder=text_encoder,
**components,
)
if not server.show_progress: