File size: 184 Bytes
db5abb4
88d11e2
807cbac
a551d94
f5ceadc
1
2
3
4
5
from transformers import LlamaForCausalLM
from .configuration_cognitivess import CognitivessConfig

class CognitivessForCausalLM(LlamaForCausalLM):
    config_class = CognitivessConfig