flax
Fix vmap axes for input_positions in dot_product_attention_with_rope
#5537
Open
Open
Fix vmap axes for input_positions in dot_product_attention_with_rope #5537
TheSaiEaranti
wants to merge
1 commit into
google:main
from
TheSaiEaranti:fix-rope-input-positions
Fix vmap axes for input_positions in dot_product_attention_with_rope
f2f0bb7c
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub