# coding=utf-8
# Copyright 2020-present the AI Algorithm Research Team.
# http://agilesoda.ai
# contact@agilesoda.ai
# Model


from transformers import PretrainedConfig
from transformers import AutoConfig


class MistralForEmbeddingConfig(PretrainedConfig):
    model_type = "mistral_for_embedding"
    
    
AutoConfig.register("mistral_for_embedding", MistralForEmbeddingConfig)