o پi@sTddlmZmZddlZddlmZmZeGdddeZeGdddeZdS)) dataclassfieldN) VAEArchConfig VAEConfigc@sXeZdZUdZeed<dZeed<dZedBed<dZeed<d Z e ed fed <d Z eed <dZ e e d fed<dZe ed fed<dZe ed<dZeed<dZeed<dZeed<dZedBed<dZeed<dZeed<dZeed<d Ze ejBed!<dZe e d fdBed"<dZe e d fdBed#<dZe dBed$<dZeed%<dZeed&<dS)'GlmImageVAEArchConfigspatial_compression_ratio`base_dimNdecoder_base_dimz_dim)rr.dim_multrnum_res_blocks attn_scales)FTTtemperal_downsamplegdropoutF is_residualinput_channels out_channels patch_sizerscale_factor_temporalscale_factor_spatialT clip_outputrscaling_factor latents_mean latents_std shift_factorlatent_channels in_channels) __name__ __module__ __qualname__rint__annotations__r r r rtuplerrfloatrboolrrrrrrrrrtorchTensorr r!r"r#r$rrrf/home/ubuntu/.local/lib/python3.10/site-packages/sglang/multimodal_gen/configs/models/vaes/glmimage.pyrs.            rc@sheZdZUeedZeed<dZeed<dZ eed<dZ eed<dZ eed<d d Z d d Z d dZdS)GlmImageVAEConfig)default_factory arch_configTuse_feature_cacheF use_tilinguse_temporal_tilinguse_parallel_tilingcCsdt|jjSNr)lenr2rselfrrr/get_vae_scale_factor0sz&GlmImageVAEConfig.get_vae_scale_factorcCs|j|jd|_dSr7)tile_sample_min_num_framestile_sample_stride_num_framesblend_num_framesr9rrr/ __post_init__3s  zGlmImageVAEConfig.__post_init__cCs$dt|jj|j_|jj|j_dSr7)r8r2rvae_scale_factorrr9rrr/ post_init8s zGlmImageVAEConfig.post_initN)r%r&r'rrr2r)r3r,r4r5r6r;r?rArrrr/r0&s      r0) dataclassesrrr-.sglang.multimodal_gen.configs.models.vaes.baserrrr0rrrr/s