PsychiatryNLPKit.model.BaseLLM

class PsychiatryNLPKit.model.BaseLLM(name, key=None)

Abstract base class for all model wrappers.

__init__(name, key=None)

Methods

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

Attributes

is_loaded	Whether the model is currently loaded in memory.
model
tokenizer

__init__(name, key=None)

property model: Any

property tokenizer: Any

property is_loaded: bool

Whether the model is currently loaded in memory.

abstractmethod load()

unload()