Skip to content

fix(nnx): Fix Linen checkpoint restoration into NNX Gemma4 model - #4619

Draft
hsuan-lun-chiang wants to merge 7 commits into
mainfrom
fix/gemma4-nnx-linen-ckpt-mapping
Draft

fix(nnx): Fix Linen checkpoint restoration into NNX Gemma4 model#4619
hsuan-lun-chiang wants to merge 7 commits into
mainfrom
fix/gemma4-nnx-linen-ckpt-mapping

Remove manual _free_device_memory call before restore to prevent dele…

f3206b2
Select commit
Loading
Failed to load commit list.