o پi @sTddlmZmZddlZddlmZmZeGdddeZeGdddeZdS)) dataclassfieldN) VAEArchConfig VAEConfigc@seZdZUdZeed<dZedBed<dZeed<dZe edfed <d Z eed <d Z e e dfed <dZ e edfed<dZe ed<dZe e dfed<dZe e dfed<dZeed<dZeed<dZeed<dZedBed<dZeed<dZeed<d Zeed!<d"d#ZdS)$WanVAEArchConfig`base_dimNdecoder_base_dimz_dim)r.dim_multr num_res_blocks attn_scales)FTTtemperal_downsamplegdropout)gy):gM Og^)gQ?gtVƿgZӼ?gBfjÿgU0*?gL F%u?gMg&†?gz6>׿g F%uȿg[ AcgMJ?gW2ıҿ latents_mean)g_L@gNё\C?gQ@g?@g9#J{?g|a2U?gHPs@g0*@gJ{/L& @gJY8@g]C@g(?gK46?gS:?go_Ι@g-? latents_stdF is_residual in_channels out_channels patch_sizerscale_factor_temporalscale_factor_spatialT clip_outputcCsTdt|jd|jddd|_t|jd|jddd|_|j|_ |j |_ dS)Ng?r ) torchtensorrviewr scaling_factorr shift_factorrtemporal_compression_ratiorspatial_compression_ratioselfrrd/home/ubuntu/.local/lib/python3.10/site-packages/sglang/multimodal_gen/configs/models/vaes/wanvae.py __post_init__As    zWanVAEArchConfig.__post_init__)__name__ __module__ __qualname__rint__annotations__r r rtuplerrfloatrboolrrrrrrrrrrr*rrrr)r s&           rc@speZdZUeedZeed<dZeed<dZ eed<dZ eed<dZ eed<dZ eed <dZ eed <d d Zd S) WanVAEConfig)default_factory arch_configTuse_feature_cacheF use_tilinguse_temporal_tilinguse_parallel_tilinguse_parallel_encodeuse_parallel_decodecCs|j|jd|_dS)Nr )tile_sample_min_num_framestile_sample_stride_num_framesblend_num_framesr'rrr)r*Xs  zWanVAEConfig.__post_init__N)r+r,r-rrr5r/r6r2r7r8r9r:r;r*rrrr)r3Ls        r3) dataclassesrrr .sglang.multimodal_gen.configs.models.vaes.baserrrr3rrrr)s@