apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): support normal LoRA keys in SDXL LoRAs

This commit is contained in:
Sean Sube 2023-11-21 19:10:11 -06:00
parent c4aa262ed7
commit 60615e097b
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
api/onnx_web/convert/diffusion/lora.py
View File

@ -86,6 +86,15 @@ def fix_xl_names(keys: Dict[str, Any], nodes: List[NodeProto]):
block = "up_blocks" block = "up_blocks"
elif root.startswith("text_model"): elif root.startswith("text_model"):
block = "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: else:
logger.warning("unknown XL key name: %s", key) logger.warning("unknown XL key name: %s", key)
fixed[key] = value fixed[key] = value