PsychiatryNLPKit.model.HFGenerativeLLM

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

Marker for autoregressive generative models.

__init__(name, key=None)[source]

Methods

__init__(name[, key])

load()

unload()

Attributes

is_loaded

Whether the model is currently loaded in memory.

model

tokenizer

__init__(name, key=None)[source]