Please publish train_state_unpadded_shape_dtype_struct

[sdmorrey]

We really need this file very soon the model will not run at all

Restored checkpoint in 6.29 seconds. Jitting decoding. ERROR:absl:For checkpoint version > 1.0, we require users to provide train_state_unpadded_shape_dtype_struct during checkpoint saving/restoring, to avoid potential silent bugs when loading checkpoints to incompatible unpadded shapes of TrainState.

[rajatsen91]

I just checked in a fresh install. Model loading and inference still works despite of that error message. Please verify whether you can run inference. If so, for now please treat this ERROR:absl as a warning not an error.

[sf19881108]

I have downloaded the model locally and encountered an error while running the code：
Constructed model weights in 1.72 seconds.
Restoring checkpoint from model/timesfm-1.0-200m/checkpoints/.
WARNING:absl:No registered CheckpointArgs found for handler type: <class 'paxml.checkpoints.FlaxCheckpointHandler'>
WARNING:absl:Configured CheckpointManager using deprecated legacy API. Please follow the instructions at https://orbax.readthedocs.io/en/latest/api_refactor.html to migrate by May 1st, 2024.
WARNING:absl:train_state_unpadded_shape_dtype_struct is not provided. We assume train_state is unpadded.
ERROR:absl:For checkpoint version > 1.0, we require users to provide
train_state_unpadded_shape_dtype_struct during checkpoint
saving/restoring, to avoid potential silent bugs when loading
checkpoints to incompatible unpadded shapes of TrainState.
Restored checkpoint in 0.95 seconds.
Jitting decoding.
Does anyone knows how to deal it?

[godcrying]

It has been 23 July. Maybe no one knows how to deal it.

[HuangY0]

Just wait for the jitting to end