o پi@sLddlmZmZddlmZmZeGdddeZeGdddeZdS)) dataclassfield) DiTArchConfig DiTConfigcseZdZUdZeed<dZeed<dZedBed<dZeed<d Z eed <d Z eed <d Z eed<dZ eed<dZ eed<dZeed<eedZeeeeefed<edddZeed<fddZZS)GlmImageArchConfig patch_size in_channelsN out_channels num_layersattention_head_dim num_attention_heads condition_dimi@ prior_vq_quantizer_codebook_sizeitext_embed_dimitime_embed_dimdefault_factorystacked_params_mappingcCsddiS)Nz9^(transformer_blocks\.\d+\.attn\..*\.lora_[AB])\.default$z\1rrrf/home/ubuntu/.local/lib/python3.10/site-packages/sglang/multimodal_gen/configs/models/dits/glmimage.pyszGlmImageArchConfig.param_names_mappingcs2t|jp |j|_|j|j|_|j|_dS)N)super __post_init__r r rr hidden_sizenum_channels_latents)self __class__rrrs  z GlmImageArchConfig.__post_init__)__name__ __module__ __qualname__rint__annotations__r r r rrrrrrrlistrtuplestrrdictr __classcell__rrr#rrs          rc@s,eZdZUeedZeed<dZe ed<dS)GlmImageDitConfigr arch_configglmimageprefixN) r%r&r'rrr0rr)r2r,rrrrr/#s r/N) dataclassesrr.sglang.multimodal_gen.configs.models.dits.baserrrr/rrrrs