759c759,760
<                 fa_forward_kwargs["window_size"] = (-1, 0) if causal else (-1, -1)
---
>                 fa_forward_kwargs["window_size_left"] = -1
>                 fa_forward_kwargs["window_size_right"] = 0 if causal else -1
1063c1064,1065
<                                     fa_forward_kwargs["window_size"] = (-1, -1)
---
>                                     fa_forward_kwargs["window_size_left"] = -1
>                                     fa_forward_kwargs["window_size_right"] = -1
1210c1212,1213
<                                     fa_forward_kwargs["window_size"] = (-1, -1)
---
>                                     fa_forward_kwargs["window_size_left"] = -1
>                                     fa_forward_kwargs["window_size_right"] = -1
1883c1886,1887
<                             fa_backward_kwargs["window_size"] = (-1, 0)
---
>                             fa_backward_kwargs["window_size_left"] = -1
>                             fa_backward_kwargs["window_size_right"] = 0
2009c2013,2014
<                             fa_backward_kwargs["window_size"] = (-1, -1)
---
>                             fa_backward_kwargs["window_size_left"] = -1
>                             fa_backward_kwargs["window_size_right"] = -1
2137c2142,2143
<                             fa_backward_kwargs["window_size"] = (-1, -1)
---
>                             fa_backward_kwargs["window_size_left"] = -1
>                             fa_backward_kwargs["window_size_right"] = -1
2232c2238,2239
<                         fa_backward_kwargs["window_size"] = (-1, -1)
---
>                         fa_backward_kwargs["window_size_left"] = -1
>                         fa_backward_kwargs["window_size_right"] = -1
2695c2702,2703
<                             fa_forward_kwargs["window_size"] = window_size_per_step[i]
---
>                             fa_forward_kwargs["window_size_left"] = window_size_per_step[i][0]
>                             fa_forward_kwargs["window_size_right"] = window_size_per_step[i][1]
2895c2903,2904
<                             fa_backward_kwargs["window_size"] = window_size_per_step[i]
---
>                             fa_backward_kwargs["window_size_left"] = window_size_per_step[i][0]
>                             fa_backward_kwargs["window_size_right"] = window_size_per_step[i][1]
3037c3046,3047
<                 fa_forward_kwargs["window_size"] = window_size
---
>                 fa_forward_kwargs["window_size_left"] = window_size[0]
>                 fa_forward_kwargs["window_size_right"] = window_size[1]
3355c3365,3366
<                 fa_backward_kwargs["window_size"] = ctx.window_size
---
>                 fa_backward_kwargs["window_size_left"] = ctx.window_size[0]
>                 fa_backward_kwargs["window_size_right"] = ctx.window_size[1]