src/pipelines/sat_loader.py

"""Shared SAT model loader for pipelines."""
from wtpsplit import SaT


def load_sat(device: str = "cpu") -> SaT:
    """Load and return the SAT-12L sentence segmentation model."""
    model = SaT("sat-12l")
    model.half().to(device)
    return model