o Û·itã@sdZddlZGdd„dƒZdS)zj TeaCache state management. This module manages the state for TeaCache hooks across diffusion timesteps. éNc@s"eZdZdZdd„Zddd„ZdS) Ú TeaCacheStatez State management for TeaCache hook. Tracks caching state across diffusion timesteps, managing counters, accumulated distances, and cached residuals for the TeaCache algorithm. cCó"d|_d|_d|_d|_d|_dS)z Initialize empty TeaCache state.rçN©ÚcntÚaccumulated_rel_l1_distanceÚprevious_modulated_inputÚprevious_residualÚprevious_residual_encoder©Úself©r ú^/home/ubuntu/vllm_env/lib/python3.10/site-packages/vllm_omni/diffusion/cache/teacache/state.pyÚ__init__s  zTeaCacheState.__init__ÚreturnNcCr)z2Reset all state variables for a new inference run.rrNrr r r rÚreset s  zTeaCacheState.reset)rN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrr r r rr s r)rÚtorchrr r r rÚs