8 lines
234 B
Python
8 lines
234 B
Python
from transformers import LlamaModel, LlamaForCausalLM
|
|
from .configuration_ilama import IlamaConfig
|
|
|
|
class IlamaModel(LlamaModel):
|
|
config_class = IlamaConfig
|
|
|
|
class IlamaForCausalLM(LlamaForCausalLM):
|
|
config_class = IlamaConfig |