From 2e0f79a396c99ecfb9142ada5885d96fdb005c2e Mon Sep 17 00:00:00 2001 From: silveroxides Date: Sat, 2 May 2026 19:25:20 +0200 Subject: [PATCH] handle Anima net. layer prefix --- nodes/lora_extract_svd.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/nodes/lora_extract_svd.py b/nodes/lora_extract_svd.py index 440fb44..b2eeb4f 100644 --- a/nodes/lora_extract_svd.py +++ b/nodes/lora_extract_svd.py @@ -692,9 +692,13 @@ def _format_lora_key(key: str) -> str: key = key[:-7] # Handle ComfyUI Checkpoint wrapper - if key.startswith("model.diffusion_model."): + if key.startswith("model.diffusion_model.") or key.startswith("net."): # Check if it's a Diffusers-format transformer inside a checkpoint - inner_key = key[22:] # len("model.diffusion_model.") + if key.startswith("model.diffusion_model."): + inner_key = key[22:] # len("model.diffusion_model.") + else: + inner_key = key[4:] # len("net.") + if inner_key.startswith("transformer_blocks") or inner_key.startswith("single_transformer_blocks"): return f"transformer.{inner_key}" # Standard checkpoint or Original Flux