cognitivess / cognitivess_model /configuration_cognitivess.py
cognitivess's picture
Update cognitivess_model/configuration_cognitivess.py
b9b2851 verified
raw
history blame
257 Bytes
from transformers import LlamaConfig
class CognitivessConfig(LlamaConfig):
model_type = "cognitivess"
def __init__(self, quantization_config=None, **kwargs):
super().__init__(**kwargs)
self.quantization_config = quantization_config