Skip to content

Commit

Permalink
RF relative_positional_encoding, ignore dyn dims in concat
Browse files Browse the repository at this point in the history
  • Loading branch information
albertz committed Dec 11, 2024
1 parent 55f6bbc commit 76b08ff
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions returnn/frontend/attention.py
Original file line number Diff line number Diff line change
Expand Up @@ -869,6 +869,7 @@ def _make_indices(
indices, out_spatial_dim = rf.concat(
(q_pos_vec - query_spatial_dim_m1.get_dim_value_tensor(), query_spatial_dim_m1),
(kv_pos_vec, key_value_spatial_dim),
handle_dynamic_dims=False,
)
if query_offset is not None:
indices = indices - query_offset
Expand Down

0 comments on commit 76b08ff

Please sign in to comment.