Checkpoints for Foundation Models

July 21, 2024 · View on GitHub

For full reproducibility, we provide the checkpoints for some foundation models as of the paper completion date.

Download the checkpoints from Google Drive with:

# By downloading, you agree to the terms of the original license agreements.
sh scripts/prepare_checkpoints.sh # in root directory

You can also download the newest checkpoints from the following repositories:

For Timer, download the checkpoints from its official repository (Google Drive or Tsinghua Cloud) under the folder ./checkpoints/timer/Timer_67M_UTSD_4G.pt.
For ForecastPFN, download the checkpoints from its official repository (Google Drive) under the folder ./checkpoints/ForecastPFN/saved_weights.
For UniTS, download the checkpoints units_x128_pretrain_checkpoint.pth from its official repository under the folder ./checkpoints/units/units_x128_pretrain_checkpoint.pth.
For Lag-Llama, download the checkpoints lag-llama.ckpt from its huggingface repository under the folder ./checkpoints/lag-llama/lag-llama.ckpt.
For other models, they can be automatically downloaded from huggingface during the first run.