o
    پi
                     @  s   d dl mZ d dlmZ d dlmZ d dlmZmZ d dl	m
Z
mZ er6d dlmZ d dlmZ d dlmZ dddZg dZdS )    )annotations)TYPE_CHECKING)nn)BaseModelLoaderget_model_loader)get_architecture_class_nameget_model_architecture)DeviceConfig)
LoadConfig)ModelConfigmodel_configr   load_configr
   device_configr	   return	nn.Modulec                 C  s   t || }|j| |dS )N)r   r   )r   
load_model)r   r   r   loader r   T/home/ubuntu/.local/lib/python3.10/site-packages/sglang/srt/model_loader/__init__.py	get_model   s
   
r   )r   r   r   r   r   N)r   r   r   r
   r   r	   r   r   )
__future__r   typingr   torchr   sglang.srt.model_loader.loaderr   r   sglang.srt.model_loader.utilsr   r    sglang.srt.configs.device_configr	   sglang.srt.configs.load_configr
   sglang.srt.configs.model_configr   r   __all__r   r   r   r   <module>   s   
