File size: 234 Bytes
33ad917
41a5529
70d3996
 
33ad917
 
c51b672
26257ee
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