o
    
۾i                     @   s&   d dl mZmZ eG dd dZdS )    )	dataclassfieldc                   @   s   e Zd ZU eedZee ed< eedZee ed< eedZ	ee ed< eedZ
ee ed< e				d
deee B deee B deee B deee B dd f
dd	ZdS )MultiModelKeys)default_factorylanguage_model	connectortower_model	generatorNreturnc                 K   s0   dd }t d|| ||||||d|S )Nc                 S   s$   | d u rg S t | tr| gS t| S )N)
isinstancestrlist)value r   ]/home/ubuntu/.local/lib/python3.10/site-packages/vllm/model_executor/models/module_mapping.pyto_list   s   z1MultiModelKeys.from_string_field.<locals>.to_list)r   r   r   r	   r   )r   )r   r   r   r	   kwargsr   r   r   r   from_string_field   s   z MultiModelKeys.from_string_field)NNNN)__name__
__module____qualname__r   r   r   r   __annotations__r   r   r	   staticmethodr   r   r   r   r   r   
   s*   
 



r   N)dataclassesr   r   r   r   r   r   r   <module>   s   