mirror of
https://github.com/vladmandic/automatic
synced 2026-08-31 09:31:00 +02:00
b37e275212
Adds pipelines/ernie/ernie_lora.py with try_load_lora/lokr/loha/oft entry points modeled on the z-image native loader, wires it into lora_load.load_safetensors, and adds 'ernieimage' to allow_native. ERNIE attention is not fused (separate to_q/to_k/to_v/to_out.0), so the loader skips the qkv-split machinery the z-image loader needs and supports all four families uniformly. Recognized prefixes are diffusion_model., transformer., and lora_unet_; PEFT lora_A/lora_B keys are normalized to lora_down/lora_up. Verified against PEFT, LoKR, and Kohya/AIT files in the wild with zero unmapped or shape-mismatched modules.