from transformers import PreTrainedModel
from transformers.models.llama.modeling_llama import LlamaForCausalLM

class CohereForCausalLM(LlamaForCausalLM):
    pass