"""Trimmed from models/tools.py: only what modeling_patchtst_fm.py imports."""


def count_parameters(model):
    total_params = sum(p.numel() for p in model.parameters())
    grad_params = sum(p.numel() for p in model.parameters() if p.requires_grad)
    no_grad_params = total_params - grad_params
    return total_params, grad_params, no_grad_params