Rahul-8799's picture
Create model.py
20345ed verified
raw
history blame contribute delete
164 Bytes
from transformers import PreTrainedModel
from transformers.models.llama.modeling_llama import LlamaForCausalLM
class CohereForCausalLM(LlamaForCausalLM):
pass