from transformers.models.mistral.modeling_mistral import MistralForCausalLM, MistralModel
from transformers.models.mistral.configuration_mistral import MistralConfig

class CiloConfig(MistralConfig):
    model_type = "cilo"

class CiloModel(MistralModel):
    config_class = CiloConfig

class CiloForCausalLM(MistralForCausalLM):
    config_class = CiloConfig