| # 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) | |