File size: 164 Bytes
20345ed
 
 
 
 
1
2
3
4
5
6
from transformers import PreTrainedModel
from transformers.models.llama.modeling_llama import LlamaForCausalLM

class CohereForCausalLM(LlamaForCausalLM):
    pass