o ¶ Ïi²ã@s4ddlZddlZdd„Zdd„Zdd„Zdd „ZdS) éNcCs^i}|D]}d | dd¡ dd¡¡}||||<qt |djd|djd g¡|d <|S) Núdiffusion_model.{}z .lora_B.biasz.diff_bz _norm.scalez_norm.set_weightzimg_in.lora_B.weightrúimg_in.lora_A.weightéz%diffusion_model.img_in.reshape_weight)ÚformatÚreplaceÚtorchÚtensorÚshape)ÚsdÚsd_outÚkÚk_to©rú0/mnt/c/Users/fbmor/ComfyUI/comfy/lora_convert.pyÚconvert_lora_bfl_controls &rcCstj |ddi¡S)NÚ lora_unet__Ú lora_unet_)ÚcomfyÚutilsÚstate_dict_prefix_replace©r rrrÚconvert_lora_wan_funsrcCsti}|D]3}||}d | dd¡ dd¡ dd¡ dd ¡ d d ¡ d d ¡ dd¡ dd¡ dd¡¡}|||<q|S)Nrz .down.weightz.lora_down.weightz .up.weightz.lora_up.weightz .qkv_lora2.z.txt_attn.qkv.z .qkv_lora1.z.img_attn.qkv.z .proj_lora1.z.img_attn.proj.z .proj_lora2.z.txt_attn.proj.z .qkv_lora.z .linear1_qkv.z .proj_lora.z .linear2.z .processor.Ú.)rr)r r r rr rrrÚconvert_uso_lorasø rcCsDd|vr d|vr t|ƒSd|vrt|ƒSd|vr d|vr t|ƒS|S)Nrz#single_blocks.0.norm.key_norm.scalez1lora_unet__blocks_0_cross_attn_k.lora_down.weightz-single_blocks.37.processor.qkv_lora.up.weightz.double_blocks.18.processor.qkv_lora2.up.weight)rrrrrrrÚ convert_lora$sr)rÚ comfy.utilsrrrrrrrrrÚs