o
    ߥi                     @   sj   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
mZ ejejejdG dd	 d	e
eeZd
S )    )BloomForCausalLM)Models)MODELS)Tasks   )MsModelMixin
TorchModel)	group_keymodule_namec                   @   s   e Zd ZdS )BloomForTextGenerationN)__name__
__module____qualname__ r   r   _/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/nlp/bloom/text_generation.pyr   
   s    r   N)transformersr   BloomForCausalLMTransformmodelscope.metainfor   modelscope.modelsr   modelscope.utils.constantr   backboner   r   register_moduletext_generationbloomr   r   r   r   r   <module>   s   
