@@ -567,6 +567,11 @@ std::string convert_diffusers_dit_to_original_flux(std::string name) {
567567 flux_name_map[block_prefix + " attn.norm_k.weight" ] = dst_prefix + " img_attn.norm.key_norm.scale" ;
568568 flux_name_map[block_prefix + " attn.norm_added_q.weight" ] = dst_prefix + " txt_attn.norm.query_norm.scale" ;
569569 flux_name_map[block_prefix + " attn.norm_added_k.weight" ] = dst_prefix + " txt_attn.norm.key_norm.scale" ;
570+ // Comfy-Org/LongCat-Image stores already-converted RMSNorm tensors as *.weight.
571+ flux_name_map[dst_prefix + " img_attn.norm.query_norm.weight" ] = dst_prefix + " img_attn.norm.query_norm.scale" ;
572+ flux_name_map[dst_prefix + " img_attn.norm.key_norm.weight" ] = dst_prefix + " img_attn.norm.key_norm.scale" ;
573+ flux_name_map[dst_prefix + " txt_attn.norm.query_norm.weight" ] = dst_prefix + " txt_attn.norm.query_norm.scale" ;
574+ flux_name_map[dst_prefix + " txt_attn.norm.key_norm.weight" ] = dst_prefix + " txt_attn.norm.key_norm.scale" ;
570575
571576 // ff
572577 flux_name_map[block_prefix + " ff.net.0.proj.weight" ] = dst_prefix + " img_mlp.0.weight" ;
@@ -605,8 +610,11 @@ std::string convert_diffusers_dit_to_original_flux(std::string name) {
605610
606611 flux_name_map[block_prefix + " attn.norm_q.weight" ] = dst_prefix + " norm.query_norm.scale" ;
607612 flux_name_map[block_prefix + " attn.norm_k.weight" ] = dst_prefix + " norm.key_norm.scale" ;
608- flux_name_map[block_prefix + " proj_out.weight" ] = dst_prefix + " linear2.weight" ;
609- flux_name_map[block_prefix + " proj_out.bias" ] = dst_prefix + " linear2.bias" ;
613+ // Comfy-Org/LongCat-Image stores already-converted RMSNorm tensors as *.weight.
614+ flux_name_map[dst_prefix + " norm.query_norm.weight" ] = dst_prefix + " norm.query_norm.scale" ;
615+ flux_name_map[dst_prefix + " norm.key_norm.weight" ] = dst_prefix + " norm.key_norm.scale" ;
616+ flux_name_map[block_prefix + " proj_out.weight" ] = dst_prefix + " linear2.weight" ;
617+ flux_name_map[block_prefix + " proj_out.bias" ] = dst_prefix + " linear2.bias" ;
610618 }
611619
612620 // --- final layers ---
0 commit comments