PsychiatryNLPKit.model.HFGenerativeLLM

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

Marker for autoregressive generative models.

__init__(name, key=None)

Methods

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

Attributes

model
tokenizer

__init__(name, key=None)