fred-dev/comfy_ui_ali
0
1�
2 ���gN � � � d dl Z d� Zd� Zy)� Nc �� � i }| D ]; }dj |j dd� j dd� � }| | ||< �= t j | d j d | d j d g� |d3< |S ) Nzdiffusion_model.{}z .lora_B.biasz.diff_bz _norm.scalez_norm.scale.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_tos �]/Users/fredrodrigues/Library/CloudStorage/Dropbox/Code/comfyui_hardware/comfy/lora_convert.py�convert_lora_bfl_controlr s� � �
�F�
��#�*�*�1�9�9�^�Y�+O�+W�+W�Xe�g� ,A� B���!�u��t� � � 7<�l�l�B�G]�D^�Dd�Dd�ef�Dg�ik� mC� jD� jJ� jJ� KL� jM� DN� 7O�F� 2�3� �M� c �, � d| v rd| v r t | � S | S )Nr z#single_blocks.0.norm.key_norm.scale)r )r s r � convert_lorar s# � � ��#�(M�QS�(S�'��+�+�
�Ir )r r r � r r �<module>r s �� ��r