PsychiatryNLPKit.model.BaseLLM

class PsychiatryNLPKit.model.BaseLLM(name, key=None)[source]

Abstract base class for all model wrappers.

__init__(name, key=None)[source]

Methods

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

Attributes

`model`
`tokenizer`

__init__(name, key=None)[source]

property model: Any

property tokenizer: Any

abstractmethod load()[source]

unload()[source]