o پi>@sLddlmZmZddlmZmZeGdddeZeGdddeZdS)) dataclassfield) VAEArchConfig VAEConfigc@seZdZUdZeed<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<dS) QwenImageVAEArchConfigspatial_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_output)__name__ __module__ __qualname__rint__annotations__r r r rtuplerrfloatrboolrrrrrrrrrrrg/home/ubuntu/.local/lib/python3.10/site-packages/sglang/multimodal_gen/configs/models/vaes/qwenimage.pyr s"          rc@sheZdZUeedZeed<dZeed<dZ eed<dZ eed<dZ eed<d d Z d d Z d dZdS)QwenImageVAEConfig)default_factory arch_configTuse_feature_cacheF use_tilinguse_temporal_tilinguse_parallel_tilingcCsdt|jjSNr)lenr*rselfrrr'get_vae_scale_factor)sz'QwenImageVAEConfig.get_vae_scale_factorcCs|j|jd|_dSr/)tile_sample_min_num_framestile_sample_stride_num_framesblend_num_framesr1rrr' __post_init__,s  z QwenImageVAEConfig.__post_init__cCs$dt|jj|j_|jj|j_dSr/)r0r*rvae_scale_factorrr1rrr' post_init1s zQwenImageVAEConfig.post_initN)rr r!rrr*r#r+r&r,r-r.r3r7r9rrrr'r(s      r(N) dataclassesrr.sglang.multimodal_gen.configs.models.vaes.baserrrr(rrrr's