`medcat.utils.relation_extraction.llama.config`

Module Contents

Classes

LlamaConfig_RelationExtraction

Class for LlamaConfig

Attributes

medcat.utils.relation_extraction.llama.config.logger

class medcat.utils.relation_extraction.llama.config.LlamaConfig_RelationExtraction(pretrained_model_name_or_path, **kwargs)

Bases: medcat.utils.relation_extraction.config.BaseConfig_RelationExtraction

Class for LlamaConfig

name = 'llama-config'

pretrained_model_name_or_path = 'meta-llama/Llama-3.1-8B'

classmethod load(pretrained_model_name_or_path, relcat_config, **kwargs)

Parameters:

pretrained_model_name_or_path (str) –
relcat_config (medcat.config_rel_cat.ConfigRelCAT) –

Return type:

LlamaConfig_RelationExtraction