fix: update config

mesakhcienet · mesakhcienet · commit 42cfb6483539 · 2026-03-18T18:03:10.000+08:00
diff --git a/src/maxtext/configs/base.yml b/src/maxtext/configs/base.yml
@@ -1086,8 +1086,8 @@ position_id_per_seconds: 25
 subslice_shape: ""
 
 # NNX
-enable_nnx: false
-pure_nnx_decoder: false
+enable_nnx: true
+pure_nnx_decoder: true
 
 ################################## Qwen3-Next Specific Configs ##################################
 # Kernel size for the 1D convolution in the Gated Delta Net
