Fix NNX ValueError in NNXDecoder#3429

copybara-service[bot] merged 1 commit intoAI-Hypercomputer:mainAI-Hypercomputer/maxtext:mainfrom

CIeNET-International:charlesli/fix_nnx_errorCIeNET-International/maxtext:charlesli/fix_nnx_errorCopy head branch name to clipboard