
    
3j                          " S  S5      r g)c                   4    \ rS rSrSS jrS rS rSS jrSrg)	AttentionBank   Nc                     UUUS.U l         g )N)
save_steps	block_mapinject_steps_data)selfr   r   r   s       T/home/wildlama/comfy/ComfyUI/custom_nodes/ComfyUI-LTXVideo/tricks/utils/attn_bank.py__init__AttentionBank.__init__   s    $"(

    c                      U R                   U   $ Nr	   )r   keys     r   __getitem__AttentionBank.__getitem__	   s    zz#r   c                      X R                   U'   g r   r	   )r   r   values      r   __setitem__AttentionBank.__setitem__   s    

3r   c                 8    U R                   R                  X5      $ r   )r
   get)r   r   defaults      r   r   AttentionBank.get   s    zz~~c++r   r	   r   )	__name__
__module____qualname____firstlineno__r   r   r   r   __static_attributes__ r   r   r   r      s    
 ,r   r   N)r   r"   r   r   <module>r#      s   , ,r   