from transformers import LlamaModel
from .configuration_ilama import IlamaConfig

class IlamaModel(LlamaModel):
    config_class = IlamaConfig