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 )
    )GPTNeoConfig)GPTNeoModel)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	   Z/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/nlp/gpt_neo/backbone.pyr      s   zGPTNeoModel.__init__)__name__
__module____qualname__r   __classcell__r	   r	   r   r   r   
   s    r   N)transformersr   r   GPTNeoModelTransformmodelscope.metainfor   modelscope.models.builderr   modelscope.utils.constantr   register_modulebackbonegpt_neor	   r	   r	   r   <module>   s   