From c8f886fe08b5ffb748fd82ae65f7eec9941427f2 Mon Sep 17 00:00:00 2001 From: Bartok9 Date: Fri, 10 Jul 2026 20:03:20 -0400 Subject: [PATCH] fix(convert): look for assets inside checkpoint_dir MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Salvage of Physical-Intelligence/openpi#661 by @fecet — rebased to main. Conversion copied assets from checkpoint_dir.parent / assets; checkpoints store assets alongside weights under checkpoint_dir / assets (#638). --- examples/convert_jax_model_to_pytorch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/convert_jax_model_to_pytorch.py b/examples/convert_jax_model_to_pytorch.py index 632c0b8782..d854fe0c98 100644 --- a/examples/convert_jax_model_to_pytorch.py +++ b/examples/convert_jax_model_to_pytorch.py @@ -533,7 +533,7 @@ def __init__(self): safetensors.torch.save_model(pi0_model, os.path.join(output_path, "model.safetensors")) # Copy assets folder if it exists - assets_source = pathlib.Path(checkpoint_dir).parent / "assets" + assets_source = pathlib.Path(checkpoint_dir) / "assets" if assets_source.exists(): assets_dest = pathlib.Path(output_path) / "assets" if assets_dest.exists():