From 421ed632b19aed00ff5482357163da8d0fb51f34 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Wed, 15 Mar 2023 08:53:15 -0500 Subject: [PATCH] only use name from first inversion --- api/onnx_web/diffusers/load.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py index cf0462a9..d03e37da 100644 --- a/api/onnx_web/diffusers/load.py +++ b/api/onnx_web/diffusers/load.py @@ -216,7 +216,7 @@ def load_pipeline( } if inversions is not None and len(inversions) > 0: - inversion = inversions[0] + inversion = inversions[0][0] logger.debug("loading Textual Inversion from %s", inversion) # TODO: blend the inversion models components["text_encoder"] = OnnxRuntimeModel.from_pretrained(