o
    ߥi                     @   sb   d dl mZ d dl mZ d dlmZ d dlmZ d dlm	Z	 ej
e	jejdG dd deZd	S )
    )
GPT2Config)	GPT2Model)Models)	BACKBONES)Tasks)	group_keymodule_namec                       s   e Zd Z fddZ  ZS )r   c                    s   t di |}t | d S )N )r   super__init__)selfkwargsconfig	__class__r	   W/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/nlp/gpt2/backbone.pyr      s   zGPT2Model.__init__)__name__
__module____qualname__r   __classcell__r	   r	   r   r   r   
   s    r   N)transformersr   r   GPT2ModelTransformmodelscope.metainfor   modelscope.models.builderr   modelscope.utils.constantr   register_modulebackbonegpt2r	   r	   r	   r   <module>   s   