o
    ٷi                     @   s   d dl mZmZmZ i dddddddd	d
ddddddddddddddddddddddd d!d"Zi eeZei d#d$ e D d%d$ e D Zd&S )'    )_VLLM_MODELS_LazyRegisteredModel_ModelRegistry#Qwen2_5OmniForConditionalGeneration)qwen2_5_omnir   r   Qwen2_5OmniThinkerModel)r   qwen2_5_omni_thinker*Qwen2_5OmniThinkerForConditionalGenerationQwen2_5OmniTalkerModel)r   qwen2_5_omni_talker)Qwen2_5OmniTalkerForConditionalGenerationQwen2_5OmniToken2WavModel)r   qwen2_5_omni_token2wav0Qwen2_5OmniToken2WavForConditionalGenerationVLLMQwen2_5OmniToken2WavDiTModel)r   r   r   Qwen2ForCausalLM_old)r   	qwen2_oldQwen2ForCausalLM$Qwen3OmniMoeForConditionalGeneration)
qwen3_omnir   r   +Qwen3OmniMoeThinkerForConditionalGeneration)r   qwen3_omni_moe_thinkerr   *Qwen3OmniMoeTalkerForConditionalGeneration)r   qwen3_omni_moe_talkerr   Qwen3OmniMoeCode2Wav)r   qwen3_omni_code2wavr    Qwen3TTSForConditionalGeneration)	qwen3_ttsqwen3_tts_talker&Qwen3TTSTalkerForConditionalGenerationr   Qwen3TTSCode2Wav)r   qwen3_tts_code2wavr    MiMoAudioModel)
mimo_audior#   !MiMoAudioForConditionalGenerationMiMoAudioLLMModel)r#   mimo_audio_llm$MiMoAudioLLMForConditionalGenerationMiMoAudioToken2WavModel)r#   mimo_audio_code2wav.MiMoAudioToken2WavForConditionalGenerationVLLM GlmImageForConditionalGeneration)	glm_imageglm_image_arr+   c                 C   s(   i | ]\}\}}|t d | |dqS )zvllm.model_executor.models.module_name
class_namer   ).0
model_archmod_relnamecls_name r6   \/home/ubuntu/.local/lib/python3.10/site-packages/vllm_omni/model_executor/models/registry.py
<dictcomp>c   s    
r8   c                 C   s0   i | ]\}\}}}|t d | d| |dqS )z vllm_omni.model_executor.models..r.   r1   )r2   r3   
mod_folderr4   r5   r6   r6   r7   r8   j   s    N)#vllm.model_executor.models.registryr   r   r   _OMNI_MODELS_VLLM_OMNI_MODELSitemsOmniModelRegistryr6   r6   r6   r7   <module>   sh    !&+05:@EJPX