search

google-research / timesfm

TimesFM (Time Series Foundation Model) is a pretrained time-series foundation model developed by Google Research for time-series forecasting.
https://research.google/blog/a-decoder-only-foundation-model-for-time-series-forecasting/
Apache License 2.0
3.53k stars 297 forks source link

can't load local checkpoint #105

Open lizhanyang505 opened 1 month ago

lizhanyang505 commented 1 month ago

Download checkpoint path : /mnt/data1/hf/hub/models--google--timesfm-1.0-200m.

(py3.10_tfm) root@a-Super-Server:~# ls -R /mnt/data1/hf/hub/models--google--timesfm-1.0-200m /mnt/data1/hf/hub/models--google--timesfm-1.0-200m: blobs refs snapshots

/mnt/data1/hf/hub/models--google--timesfm-1.0-200m/blobs: 15ec06abe1a5fb6da2e163feadd566197a1e7af1 baf22ce8251e40ee31c7d48f283645072c26fe12 ff56f80e247a83a91ae67db6df5a19b67760ed2e 4266fd55f80e37feabf896a93f32ab871b9f7afc d147f95bbe40ffeb1e8d61643925dd29eeb00b5696db33bf9f6e1887682f6391

/mnt/data1/hf/hub/models--google--timesfm-1.0-200m/refs: main

/mnt/data1/hf/hub/models--google--timesfm-1.0-200m/snapshots: 8775f7531211ac864b739fe776b0b255c277e2be

/mnt/data1/hf/hub/models--google--timesfm-1.0-200m/snapshots/8775f7531211ac864b739fe776b0b255c277e2be: checkpoints README.md

/mnt/data1/hf/hub/models--google--timesfm-1.0-200m/snapshots/8775f7531211ac864b739fe776b0b255c277e2be/checkpoints: checkpoint_1100000

/mnt/data1/hf/hub/models--google--timesfm-1.0-200m/snapshots/8775f7531211ac864b739fe776b0b255c277e2be/checkpoints/checkpoint_1100000: descriptor metadata state

/mnt/data1/hf/hub/models--google--timesfm-1.0-200m/snapshots/8775f7531211ac864b739fe776b0b255c277e2be/checkpoints/checkpoint_1100000/descriptor: descriptor.pbtxt

/mnt/data1/hf/hub/models--google--timesfm-1.0-200m/snapshots/8775f7531211ac864b739fe776b0b255c277e2be/checkpoints/checkpoint_1100000/metadata: metadata

/mnt/data1/hf/hub/models--google--timesfm-1.0-200m/snapshots/8775f7531211ac864b739fe776b0b255c277e2be/checkpoints/checkpoint_1100000/state: checkpoint

local_checkpoint_path = "/mnt/data1/hf/hub/models--google--timesfm-1.0-200m/snapshots/8775f7531211ac864b739fe776b0b255c277e2be/checkpoints/checkpoint_1100000/state/"

tfm = timesfm.TimesFm( context_len=480, horizon_len=14, input_patch_len=32, output_patch_len=128, num_layers=20, model_dims=1280, backend="gpu", )

checkpoint_path=local_checkpoint_path,

tfm.load_from_checkpoint(checkpoint_path=local_checkpoint_path, repo_id="google/timesfm-1.0-200m")

ValueError: No checkpoints were found in directory checkpoint_dir=PosixGPath('/mnt/data1/hf/hub/models--google--timesfm-1.0-200m/snapshots/8775f7531211ac864b739fe776b0b255c277e2be/checkpoints/checkpoint_1100000/state')

lizhanyang505 commented 1 month ago

image

rajatsen91 commented 1 month ago

Can you try with:

local_checkpoint_path = "/mnt/data1/hf/hub/models--google--timesfm-1.0-200m/snapshots/8775f7531211ac864b739fe776b0b255c277e2be/checkpoints i.e removing the last bits of the path?

lizhanyang505 commented 1 month ago

Thanks!!! success image