File size: 127 Bytes
db5abb4
807cbac
a551d94
f5ceadc
1
2
3
4
from transformers import LlamaForCausalLM

class CognitivessForCausalLM(LlamaForCausalLM):
    config_class = CognitivessConfig