apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

set model path for VAE sessions

This commit is contained in:
Sean Sube 2023-09-12 07:32:29 -05:00
parent d4b013068d
commit 4ccdedba89
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -376,6 +376,7 @@ def load_pipeline(
provider=device.ort_provider("vae"),
sess_options=device.sess_options(),
)
components["vae_decoder_session"]._model_path = vae_decoder
logger.debug("loading VAE encoder from %s", vae_encoder)
components["vae_encoder_session"] = OnnxRuntimeModel.load_model(
@ -383,6 +384,7 @@ def load_pipeline(
provider=device.ort_provider("vae"),
sess_options=device.sess_options(),
)
components["vae_encoder_session"]._model_path = vae_encoder
else:
logger.debug("loading VAE decoder from %s", vae_decoder)