cognitivess / cognitivess_model /modeling_cognitivess.py
cognitivess's picture
Update cognitivess_model/modeling_cognitivess.py
db5abb4 verified
raw
history blame
127 Bytes
from transformers import LlamaForCausalLM
class CognitivessForCausalLM(LlamaForCausalLM):
config_class = CognitivessConfig