33ad917
41a5529
70d3996
33ad917
c51b672
26257ee |
|
from transformers import LlamaModel, LlamaForCausalLM
from .configuration_ilama import IlamaConfig
class IlamaModel(LlamaModel):
config_class = IlamaConfig
class IlamaForCausalLM(LlamaForCausalLM):
config_class = IlamaConfig |