LLM/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2025-12-06 12:34:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

[lora] support more ZImage LoRAs (#12790)

Browse Source 
up

Co-authored-by: Álvaro Somoza <asomoza@users.noreply.github.com>
This commit is contained in:
Sayak Paul
2025-12-05 03:01:11 +08:00
committed by GitHub
parent cd00ba685b
commit 7de51b826c
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/diffusers/loaders/lora_conversion_utils.py
View File

@@ -2417,6 +2417,17 @@ def _convert_non_diffusers_z_image_lora_to_diffusers(state_dict):
state_dict = {convert_key(k): v for k, v in state_dict.items()}
def normalize_out_key(k: str) -> str:
if ".to_out" in k:
return k
return re.sub(
r"\.out(?=\.(?:lora_down|lora_up)\.weight$|\.alpha$)",
".to_out.0",
k,
)
state_dict = {normalize_out_key(k): v for k, v in state_dict.items()}
has_default = any("default." in k for k in state_dict)
if has_default:
state_dict = {k.replace("default.", ""): v for k, v in state_dict.items()}