PsychiatryNLPKit.model.HuggingFaceLLM

class PsychiatryNLPKit.model.HuggingFaceLLM(name, key=None, model_class=<class 'transformers.models.auto.modeling_auto.AutoModel'>)[source]

Concrete HuggingFace model wrapper with lazy loading.

__init__(name, key=None, model_class=<class 'transformers.models.auto.modeling_auto.AutoModel'>)[source]

Methods

`__init__`(name[, key, model_class])
`load`()
`unload`()

Attributes

`model`
`tokenizer`

__init__(name, key=None, model_class=<class 'transformers.models.auto.modeling_auto.AutoModel'>)[source]

load()[source]

unload()[source]