From adb8b191a925b5515de597c0a684d41b9c337e9a Mon Sep 17 00:00:00 2001 From: Franz Srambical <79149449+emergenz@users.noreply.github.com> Date: Wed, 24 Sep 2025 11:41:53 +0200 Subject: [PATCH] chore: more specific migration guide description --- docs_nnx/migrating/nnx_010_to_nnx_011.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs_nnx/migrating/nnx_010_to_nnx_011.rst b/docs_nnx/migrating/nnx_010_to_nnx_011.rst index 8e43759cd..eefdc1d35 100644 --- a/docs_nnx/migrating/nnx_010_to_nnx_011.rst +++ b/docs_nnx/migrating/nnx_010_to_nnx_011.rst @@ -88,7 +88,7 @@ from the ``__init__`` method and use ``Rngs.fork`` directly before passing the R Loading Checkpoints with RNGs ================================================== -When loading checkpoints in the new version, you need to drop the old RNGs structure and +When loading checkpoints from an older version in the new version, you need to drop the old RNGs structure and partially reinitialize the model with new RNGs. To do this, you can use ``nnx.jit`` to 1. Remove the RNGs from the checkpoint.