Model Module (PsychiatryNLPKit.model)

Wrappers for HuggingFace language models and multimodal embedding models with automatic device detection (CUDA / MPS / XPU / CPU) and lazy loading.

Attach model instances to a TextData object to enable analyses that require them: embedding models for similarity/density, generative models for perplexity, etc.

Language Model Wrappers

PsychiatryNLPKit.model.BaseLLM(name[, key])	Abstract base class for all model wrappers.
PsychiatryNLPKit.model.HuggingFaceLLM(name)	Concrete HuggingFace model wrapper with lazy loading.
PsychiatryNLPKit.model.HFEmbeddingLLM(name)	Marker for embedding models (sentence/word encoders).
PsychiatryNLPKit.model.HFGenerativeLLM(name)	Marker for autoregressive generative models.
PsychiatryNLPKit.model.HFMaskFillingModel(name)	Marker for masked language models.

Vision Model Wrappers

PsychiatryNLPKit.model.HuggingFaceViTModel(name)	Wrapper for SentenceTransformer multimodal embedding models.
PsychiatryNLPKit.model.HFMultimodalEmbeddingModel(name)	Marker for multimodal embedding models that share a SentenceTransformer backend.