o پi@sLddlmZmZddlmZmZeGdddeZeGdddeZdS)) dataclassfield) VAEArchConfig VAEConfigc@seZdZUdZeed<dZeed<dZeed<dZe e dfed<d Z e e dfed <d Z e edfed <d Z eed<dZe ed<dZeed<dZeed<dZeed<dZeed<dZeed<ddZdS)HunyuanVAEArchConfig in_channels out_channelslatent_channels)HunyuanVideoDownBlock3Dr r r .down_block_types)HunyuanVideoUpBlock3Drrrup_block_types)rblock_out_channelslayers_per_blocksiluact_fn norm_num_groupsg>I?scaling_factorspatial_compression_ratiotemporal_compression_ratioTmid_block_add_attentioncCsdt|jd|_dS)Nr)lenrr)selfr#h/home/ubuntu/.local/lib/python3.10/site-packages/sglang/multimodal_gen/configs/models/vaes/hunyuanvae.py __post_init__#sz"HunyuanVAEArchConfig.__post_init__N)__name__ __module__ __qualname__rint__annotations__r r r tuplestrrrrrrrfloatrrrboolr%r#r#r#r$r s           rc@s eZdZUeedZeed<dS)HunyuanVAEConfig)default_factory arch_configN)r&r'r(rrr1rr*r#r#r#r$r/'s r/N) dataclassesrr.sglang.multimodal_gen.configs.models.vaes.baserrrr/r#r#r#r$s