o
    ॵi                     @   sV   d dl mZ d dlmZ d dlmZmZ edZejded dded	e	fd
dZ
dS )    )DistributedDataParallel)
ConfigDict)Registrybuild_from_cfgparallelr   )module_name
module_clsNcfgdefault_argsc                 C   s   t | t|dS )z build parallel

    Args:
        cfg (:obj:`ConfigDict`): config dict for parallel object.
        default_args (dict, optional): Default initialization arguments.
    )r
   )r   PARALLEL)r	   r
    r   X/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/trainers/parallel/builder.pybuild_parallel   s   r   )N)torch.nn.parallel.distributedr   modelscope.utils.configr   modelscope.utils.registryr   r   r   register_moduledictr   r   r   r   r   <module>   s   