From 60615e097bcbcf0c8c0142ba69cf1301a0932361 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Tue, 21 Nov 2023 19:10:11 -0600 Subject: [PATCH] fix(api): support normal LoRA keys in SDXL LoRAs --- api/onnx_web/convert/diffusion/lora.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/api/onnx_web/convert/diffusion/lora.py b/api/onnx_web/convert/diffusion/lora.py index f99d1ee8..91904e11 100644 --- a/api/onnx_web/convert/diffusion/lora.py +++ b/api/onnx_web/convert/diffusion/lora.py @@ -86,6 +86,15 @@ def fix_xl_names(keys: Dict[str, Any], nodes: List[NodeProto]): block = "up_blocks" elif root.startswith("text_model"): block = "text_model" + elif root.startswith("down_blocks"): + fixed[fix_node_name(key)] = value + continue + elif root.startswith("mid_blocks"): + fixed[fix_node_name(key)] = value + continue + elif root.startswith("up_blocks"): + fixed[fix_node_name(key)] = value + continue else: logger.warning("unknown XL key name: %s", key) fixed[key] = value