Function _resolve_moe_active_params

src/whichllm/models/fetcher.py:171–187

Resolve active params from curated data or A*B naming hints.

(
    total_params: int,
    *model_refs: str | None,
)

def _resolve_moe_active_params(
    total_params: int,
    *model_refs: str | None,
) -> int | None:
    """Resolve active params from curated data or A*B naming hints."""
    for ref in model_refs:
        if not ref:
            continue
        active = _lookup_curated_count(_KNOWN_MOE_ACTIVE_PARAMS, ref)
        if active and active > 0:
            return active

    for ref in model_refs:
        active = _extract_active_size_hint_from_id(ref)
        if active and active > 0 and (total_params <= 0 or active < total_params):
            return active
    return None


# Sliding-window-attention (SWA) registry. We only model SWA KV-cache savings