Function token_window

tests/vibevoice/vibevoice_python_warm_bench.py:222–227

(tokens: list[int])

def token_window(tokens: list[int]) -> dict[str, Any]:
    return {
        "head": tokens[:128],
        "tail": tokens[-32:],
        "count": len(tokens),
    }


def extract_audio(outputs: Any, index: int = 0) -> np.ndarray: