Model Module (PsychiatryNLPKit.model)
Wrappers for HuggingFace language models and Vision Transformers 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
|
Abstract base class for all model wrappers. |
Concrete HuggingFace model wrapper with lazy loading. |
|
Marker for embedding models (sentence/word encoders). |
|
Marker for autoregressive generative models. |
|
Marker for masked language models. |
Vision Model Wrappers
Wrapper for SigLIP/CLIP-style vision-language models. |
|
Marker for multimodal embedding models (e.g. CLIP, SigLIP). |