File size: 234 Bytes
923c85f |
1 2 3 4 5 6 7 8 |
from transformers import LlamaModel, LlamaForCausalLM
from .configuration_ilama import IlamaConfig
class IlamaModel(LlamaModel):
config_class = IlamaConfig
class IlamaForCausalLM(LlamaForCausalLM):
config_class = IlamaConfig |