Starstruk Logo MarkStarstruk

v1.2.6

antgroup/echomimicv1.2.6Dec 31, 2024by rajatsen91

AI Summary

This release updates the TimesFM library to support the new 2.0 models, setting the median head as the default point forecaster.

Key Highlights

  • Support for TimesFM-2.0 models
  • Median head set as default point forecaster
  • Hparam support for TimesFM-2.0
  • PyTorch decoding bug fixes

New Features

  • TimesFM-2.0 model support
  • Median head default

Full Release Notes

Changes:
----
1. Add support for TimesFM-2.0 models.
2. Set the median head as the default point forecaster.

PyPI Release:
----
v1.2.6: Support for TimesFM-2.0 models.
- Add hparam support for TimesFM-2.0 models.
- Some bug fixes in pytorch decoding.
- Right now we do not support cached decoding in both jax and pytorch.

Checkpoints:
----
The TimesFM-2.0 checkpoints are available on Hugging Face:
- https://huggingface.co/google/timesfm-2.0-500m-jax
- https://huggingface.co/google/timesfm-2.0-500m-pytorch

Full Changelog:
----
https://github.com/google-research/timesfm/commits/v1.2.6