File size: 257 Bytes
cab4193
 
 
b9b2851
cab4193
b9b2851
cab4193
b9b2851
1
2
3
4
5
6
7
8
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