v1.2.6

TimmyOVO/deepseek-ocr.rsv1.2.6Dec 31, 2024by rajatsen91

AI Summary

Adds support for the new TimesFM-2.0 models and updates the default point forecaster to the median head, while fixing some PyTorch decoding bugs.

Key Highlights

  • Support for TimesFM-2.0 models (JAX and PyTorch)
  • Median head set as the default point forecaster
  • Hyperparameter support added for TimesFM-2.0
  • Bug fixes in PyTorch decoding

Breaking Changes

  • Cached decoding is no longer supported in both JAX and PyTorch
  • Default point forecaster changed to median head

New Features

  • TimesFM-2.0 model integration
  • Hyperparameter configuration for TimesFM-2.0
  • TimesFM-2.0 checkpoint availability on Hugging Face

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