o
    ߥi                      @   sn   d dl mZmZ d dlmZ d dlmZ d dlmZ	 d dl
mZ e	 ZejejejdG dd deZd	S )
    )HeadsModels)MODELS)ModelForFillMask)logger)Tasks)module_namec                   @   s   e Zd ZejZejZdS )BertForMaskedLMN)	__name__
__module____qualname__r   bertbase_model_typer   bert_mlm	head_type r   r   X/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/nlp/bert/fill_mask.pyr	      s    
r	   N)modelscope.metainfor   r   modelscope.models.builderr   +modelscope.models.nlp.task_models.fill_maskr   modelscope.utilsr   loggingmodelscope.utils.constantr   
get_loggerregister_module	fill_maskr   r	   r   r   r   r   <module>   s   