## Checkpoint Loading Error in TimesFM

[nmgvv]

Description

When attempting to load a checkpoint for the TimesFM model, I encountered the following error:

The error suggests that for checkpoint versions > 1.0, a train_state_unpadded_shape_dtype_struct parameter is required during checkpoint loading. However, I couldn't find any documentation or examples in the TimesFM repository addressing this issue or demonstrating how to provide this parameter.

Environment

• TimesFM version: [1.1.0]
• JAX version: [0.4.26]
• Flax version: [0.8.2]
• Python version: [3.10.12]

Steps to Reproduce

1. [Provide steps to reproduce the error]
2. [Include relevant code snippets if possible]

Questions

1. Is this a known issue with recent versions of TimesFM or its dependencies?
2. How can I provide the train_state_unpadded_shape_dtype_struct parameter when loading a checkpoint?
3. Are there any plans to update the documentation or examples to address this new requirement?
4. If this is related to a change in the underlying libraries (JAX/Flax), are there any compatibility notes or migration guides available?

Additional Context

I've checked the latest documentation and example code in the TimesFM repository but couldn't find any information about this new checkpoint loading requirement. Any guidance on resolving this issue or updating the checkpoint loading process would be greatly appreciated.

[siriuz42]

It's shown as an error but should not block the model loading. Can you wait for another 2 minutes or so past the error is prompted?