Fix vocab tiling memory space mismatch under parameter host offload#4533

huytransformer wants to merge 1 commit into

AI-Hypercomputer:mainAI-Hypercomputer/maxtext:mainfrom

huytransformer:htn/vocab-tiling-param-host-offloadhuytransformer/maxtext:htn/vocab-tiling-param-host-offloadCopy head branch name to clipboard