Function _swa_arch_key

src/whichllm/models/fetcher.py:235–256

Identify the SWA architecture key for a model, or None if not honored. Relies on authoritative metadata only — the raw HF config ``model_type``/ ``architectures`` (not the normalized architecture string, which collapses gemma2 and gemma3 into "gemma") and the GGUF metadata architecture.

(config: dict, model_id: str, gguf_arch: str | None)

def _swa_arch_key(config: dict, model_id: str, gguf_arch: str | None) -> str | None:
    """Identify the SWA architecture key for a model, or None if not honored.

    Relies on authoritative metadata only — the raw HF config ``model_type``/
    ``architectures`` (not the normalized architecture string, which collapses
    gemma2 and gemma3 into "gemma") and the GGUF metadata architecture. When
    none of these are present the model is left unhonored (full-context
    estimate) rather than guessed at from the model id, since a false positive
    would under-count VRAM.
    """
    model_type = config.get("model_type")
    key = _swa_key_from_arch(model_type if isinstance(model_type, str) else None)
    if key:
        return key

    arch_list = config.get("architectures") or []
    if arch_list and isinstance(arch_list[0], str):
        key = _swa_key_from_arch(arch_list[0])
        if key:
            return key

    return _swa_key_from_arch(gguf_arch)


def _resolve_sliding_window(