| from transformers import PreTrainedModel | |
| from transformers.models.llama.modeling_llama import LlamaForCausalLM | |
| class CohereForCausalLM(LlamaForCausalLM): | |
| pass | |
| from transformers import PreTrainedModel | |
| from transformers.models.llama.modeling_llama import LlamaForCausalLM | |
| class CohereForCausalLM(LlamaForCausalLM): | |
| pass | |