def __init__()

in src/models/struxgpt_base.py [0:0]

• 12 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self, cfg_path: str, **kwargs):
        cfg = load_auto(cfg_path)

        self.prompt_system_path: str = cfg['model']['prompt_system_path']
        self.prompt_template_path: str = cfg['model']['prompt_template_path']
        self.load_prompt_template()

        self.sent_token_spacy: bool = cfg['data']['sent_token_spacy']
        self.build_sent_tokenizer()

        self.scfg = StructConfig(**cfg['struct_keys'])

        self.model_name_or_path = kwargs.get('model_name_or_path', cfg['model']['model_name_or_path'])

        super().__init__(self.model_name_or_path, cfg['model']['model_type'],
                         cfg['model']['max_output_length'], cfg['model']['use_vllm'], 
                         self.prompt_system, **kwargs)