from transformers import PretrainedConfig
import torch.nn as nn
from typing import List, Union


class ChimeraConfig(PretrainedConfig):
    model_type = "marqo-chimera-arctic-bge-m"

    def __init__(self, **kwargs):
        super().__init__(**kwargs)