wanstreamer/__init__.py

"""Corrected block-causal streaming implementation for Wan2.1.

See PROGRESS.md for the measurements behind each design decision.
"""
from .rope import RopeTable, apply_rope
from .kvcache import StreamingKVCache
from .core import (timestep_to_train_scale, ModulationCache, block_forward,
                   frame_forward, sequence_forward, make_rope_table, make_cache,
                   latent_geometry)

__all__ = ['RopeTable', 'apply_rope', 'StreamingKVCache',
           'timestep_to_train_scale', 'ModulationCache', 'block_forward',
           'frame_forward', 'sequence_forward', 'make_rope_table', 'make_cache',
           'latent_geometry']