Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

can't find adapter checkpoint #201

Open
Y2502274817 opened this issue Dec 23, 2024 · 0 comments
Open

can't find adapter checkpoint #201

Y2502274817 opened this issue Dec 23, 2024 · 0 comments

Comments

@Y2502274817
Copy link

I attempted to load checkpoint ,but I can't find where to download the checkpoint

load_adapter_checkpoint(
    model=tfm,
    adapter_checkpoint_path="./checkpoints/run_20240716_163900_lyo4psz3",
    lora_rank=8,
    lora_target_modules="all",
    use_dora=True,
)

Below are the details:

TimesFM v1.2.0. See https://github.com/google-research/timesfm/blob/master/README.md for updated APIs.
Loaded Jax TimesFM.
Fetching 5 files: 100%|██████████████████████████████████████████████████████████████████████████████████████████████| 5/5 [00:00<00:00, 36408.89it/s]
Multiprocessing context has already been set.
Constructing model weights.
Constructed model weights in 2.58 seconds.
Restoring checkpoint from /home/ubuntu/.cache/huggingface/hub/models--google--timesfm-1.0-200m/snapshots/8775f7531211ac864b739fe776b0b255c277e2be/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 1.38 seconds.
Jitting decoding.
Jitted decoding in 19.11 seconds.
Fetching 5 files: 100%|██████████████████████████████████████████████████████████████████████████████████████████████| 5/5 [00:00<00:00, 45294.86it/s]
Multiprocessing context has already been set.
Constructing model weights.
Constructed model weights in 2.45 seconds.
Restoring checkpoint from /home/ubuntu/.cache/huggingface/hub/models--google--timesfm-1.0-200m/snapshots/8775f7531211ac864b739fe776b0b255c277e2be/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.62 seconds.
Jitting decoding.
Jitted decoding in 15.00 seconds.
Restoring adapter checkpoint from ./checkpoints/run_20240716_163900_lyo4psz3.
Traceback (most recent call last):
  File "/home/ubuntu/timesfm_model/timesfm/src/adapter.py", line 65, in <module>
    load_adapter_checkpoint(
  File "/home/ubuntu/timesfm_model/timesfm/src/adapter/utils.py", line 162, in load_adapter_checkpoint
    adapter_train_state = checkpoints.restore_checkpoint(
  File "/home/ubuntu/anaconda3/envs/timesfmpax/lib/python3.10/site-packages/paxml/checkpoints.py", line 221, in restore_checkpoint
    raise ValueError(f'{checkpoint_dir=!r} does not exist')
ValueError: checkpoint_dir=PosixGPath('checkpoints/run_20240716_163900_lyo4psz3') does not exist

I want to know where to download the adapter checkpoint.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant