o io@s0ddlmZmZddlmZGdddeZdS))PretrainedConfigSiglipVisionConfig) Qwen2ConfigcseZdZdZdZ          dd ed ed eeBdBdeeBdBdedBde de de de dede ffdd Z e de fddZZS) BagelConfigz$Configuration class for BAGEL model.bagelTN Fgelu_pytorch_tanhF? visual_gen visual_und llm_config vit_config vae_configlatent_patch_sizemax_latent_sizevit_max_num_patch_per_side connector_actinterpolate_postimestep_shiftc  stjdi| ||_||_t|trtdi||_n|p!t|_t|tr1tdi||_ n|p5t|_ |p=ddd|_ ||_ ||_ ||_ | |_| |_| |_dS)N) z_channels downsample)super__init__r r isinstancedictrrrrrrrrrrr) selfr r rrrrrrrrrkwargs __class__r[/home/ubuntu/vllm_env/lib/python3.10/site-packages/vllm/transformers_utils/configs/bagel.pyr s      zBagelConfig.__init__returncCs|jjS)z-Return the hidden size of the language model.)r hidden_size)r rrr$r&2szBagelConfig.hidden_size) TTNNNrrr r Fr )__name__ __module__ __qualname____doc__ model_typeboolrrrintstrfloatrpropertyr& __classcell__rrr"r$rsN      &rN) transformersrrtransformers.models.qwen2rrrrrr$s