diff --git a/api/onnx_web/diffusion/load.py b/api/onnx_web/diffusion/load.py index 57692c6f..4f1e5dd2 100644 --- a/api/onnx_web/diffusion/load.py +++ b/api/onnx_web/diffusion/load.py @@ -1,4 +1,5 @@ from logging import getLogger +from os import path from typing import Any, Optional, Tuple import numpy as np @@ -189,10 +190,9 @@ def load_pipeline( if inversion is not None: logger.debug("loading text encoder from %s", inversion) text_encoder = OnnxRuntimeModel.from_pretrained( - inversion, + path.join(inversion, "text_encoder"), provider=device.ort_provider(), sess_options=device.sess_options(), - subfolder="text_encoder", ) pipe = pipeline.from_pretrained( diff --git a/api/params.json b/api/params.json index e45aefdc..8303b47d 100644 --- a/api/params.json +++ b/api/params.json @@ -60,6 +60,10 @@ "max": 1024, "step": 8 }, + "inversion": { + "default": "", + "keys": [] + }, "left": { "default": 0, "min": 0,