Skip to content

Fix incorrect batch-dim shape annotations on vmapped _get_new_text_tokens_inner - #746

Open
PratikDhanave wants to merge 1 commit into
google-deepmind:mainfrom
PratikDhanaveFork:fix/token-utils-inner-shape-annotations
Open

Fix incorrect batch-dim shape annotations on vmapped _get_new_text_tokens_inner#746
PratikDhanave wants to merge 1 commit into
google-deepmind:mainfrom
PratikDhanaveFork:fix/token-utils-inner-shape-annotations

Fix incorrect batch-dim shape annotations on vmapped _get_new_text_to…

b925d5a
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Jul 25, 2026 in 7s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

b925d5a Author: @PratikDhanave <i.pr********ave​@gmail.com>

(Only the first commit for a unique contributor is listed.)