File size: 233 Bytes
33ad917
41a5529
70d3996
 
33ad917
 
 
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