from transformers import LlamaModel, LlamaForCausalLM
from .configuration_ilama import IlamaConfig

class IlamaModel(LlamaModel):
    config_class = IlamaConfig

class IlamaForCausalLM(LlamaForCausalLM):
    config_class = IlamaConfig