Unify token ids to be u32 in both embedding lookup and sampling#616

uuuvn merged 1 commit into

maintrymirai/uzu:mainfrom

unify_token_ids_dtypetrymirai/uzu:unify_token_ids_dtypeCopy head branch name to clipboard