o
    wi                     @   s   d dl mZ d dlmZmZmZmZmZm	Z	m
Z
mZ dejfddZdejfddZdejfdd	Zdejfd
dZdejfddZdejfddZdejfddZg dZdS )    N)Qwen2VLConfig2BQwen2VLConfig7BQwen2VLConfig72BQwen2VLModelQwen25VLConfig3BQwen25VLConfig7BQwen25VLConfig32BQwen25VLConfig72Breturnc                   C      t t ddS Nzqwen2-vlmodel_version)r   r    r   r   c/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/nemo/collections/vlm/qwen2vl/model/api.py
qwen2vl_2b      r   c                   C   r   r   )r   r   r   r   r   r   
qwen2vl_7b"   r   r   c                   C   r   r   )r   r   r   r   r   r   qwen2vl_72b'   r   r   c                   C   r   Nz	qwen25-vlr   )r   r   r   r   r   r   qwen25vl_3b,   r   r   c                   C   r   r   )r   r   r   r   r   r   qwen25vl_7b1   r   r   c                   C   r   r   )r   r   r   r   r   r   qwen25vl_32b6   r   r   c                   C   r   r   )r   r	   r   r   r   r   qwen25vl_72b;   r   r   )r   r   r   r   r   r   r   )lightning.pytorchpytorchpl"nemo.collections.vlm.qwen2vl.modelr   r   r   r   r   r   r   r	   LightningModuler   r   r   r   r   r   r   __all__r   r   r   r   <module>   s   (