o پi@sXddlmZmZddlmZddlmZmZeGdddeZeGdddeZ dS) ) dataclassfield)Tuple) DiTArchConfig DiTConfigcseZdZUdZeed<dZeed<dZedBed<dZeed<d Z eed <d Z eed <d Z eed<dZ eed<dZ eed<dZeed<dZeeeefed<eedZeeeeefed<edddZeed<fddZZS)FluxArchConfig patch_size@ in_channelsN out_channels num_layers&num_single_layersattention_head_dimnum_attention_headsijoint_attention_dimipooled_projection_dimFguidance_embeds)8raxes_dims_ropedefault_factorystacked_params_mappingcCsziddddddddd d d d d dddddddddddddddddddd d!d"d#d$d%d&d'S)(Nz^transformer\.(\w*)\.(.*)$z\1.\2z*^transformer_blocks\.(\d+)\.mlp_fc1\.(.*)$z&transformer_blocks.\1.ff.net.0.proj.\2z*^transformer_blocks\.(\d+)\.mlp_fc2\.(.*)$z!transformer_blocks.\1.ff.net.2.\2z2^transformer_blocks\.(\d+)\.mlp_context_fc1\.(.*)$z.transformer_blocks.\1.ff_context.net.0.proj.\2z2^transformer_blocks\.(\d+)\.mlp_context_fc2\.(.*)$z)transformer_blocks.\1.ff_context.net.2.\2z+^transformer_blocks\.(\d+)\.qkv_proj\.(.*)$z$transformer_blocks.\1.attn.to_qkv.\2z3^transformer_blocks\.(\d+)\.qkv_proj_context\.(.*)$z*transformer_blocks.\1.attn.to_added_qkv.\2z+^transformer_blocks\.(\d+)\.out_proj\.(.*)$z&transformer_blocks.\1.attn.to_out.0.\2z3^transformer_blocks\.(\d+)\.out_proj_context\.(.*)$z(transformer_blocks.\1.attn.to_add_out.\2z)^transformer_blocks\.(\d+)\.norm_q\.(.*)$z$transformer_blocks.\1.attn.norm_q.\2z)^transformer_blocks\.(\d+)\.norm_k\.(.*)$z$transformer_blocks.\1.attn.norm_k.\2z/^transformer_blocks\.(\d+)\.norm_added_q\.(.*)$z*transformer_blocks.\1.attn.norm_added_q.\2z/^transformer_blocks\.(\d+)\.norm_added_k\.(.*)$z*transformer_blocks.\1.attn.norm_added_k.\2z5^transformer_blocks\.(\d+)\.attn\.add_qkv_proj\.(.*)$z2^single_transformer_blocks\.(\d+)\.qkv_proj\.(.*)$z+single_transformer_blocks.\1.attn.to_qkv.\2z2^single_transformer_blocks\.(\d+)\.out_proj\.(.*)$z-single_transformer_blocks.\1.attn.to_out.0.\2z0^single_transformer_blocks\.(\d+)\.norm_q\.(.*)$z+single_transformer_blocks.\1.attn.norm_q.\2z+single_transformer_blocks.\1.attn.norm_k.\2z\1.smooth_factor_origz\1.smooth_factorz \1.proj_downz \1.proj_up)z0^single_transformer_blocks\.(\d+)\.norm_k\.(.*)$z^(.*)\.smooth_orig$z^(.*)\.smooth$z^(.*)\.lora_down$z^(.*)\.lora_up$rrrb/home/ubuntu/.local/lib/python3.10/site-packages/sglang/multimodal_gen/configs/models/dits/flux.pysR     zFluxArchConfig.param_names_mappingcs2t|jp |j|_|j|j|_|j|_dS)N)super __post_init__r r rr hidden_sizenum_channels_latents)self __class__rrr#;s  zFluxArchConfig.__post_init__)__name__ __module__ __qualname__r int__annotations__r r rrrrrrrboolrrrlistrtuplestrr!dictr# __classcell__rrr'rr s"           rc@s,eZdZUeedZeed<dZe ed<dS) FluxConfigr arch_configFluxprefixN) r)r*r+rrr5rr-r7r1rrrrr4Bs r4N) dataclassesrrtypingr.sglang.multimodal_gen.configs.models.dits.baserrrr4rrrrs 7