Skip to content

Fix vmap axes for input_positions in dot_product_attention_with_rope - #5537

Open
TheSaiEaranti wants to merge 1 commit into
google:mainfrom
TheSaiEaranti:fix-rope-input-positions
Open

Fix vmap axes for input_positions in dot_product_attention_with_rope#5537
TheSaiEaranti wants to merge 1 commit into
google:mainfrom
TheSaiEaranti:fix-rope-input-positions

Fix vmap axes for input_positions in dot_product_attention_with_rope

f2f0bb7
Select commit
Loading
Failed to load commit list.
Sign in for the full log view

The logs for this run have expired and are no longer available.