fix(qwen3-next): make activation offload work under the nested block scan - #4990

fix(qwen3-next): make activation offload work under the nested block scan#4990

NuojCheng wants to merge 1 commit into

mainAI-Hypercomputer/maxtext:mainfrom

chengnuojin-bharatgen-offloadAI-Hypercomputer/maxtext:chengnuojin-bharatgen-offloadCopy head branch name to clipboard