We are happy to announce the Uni2TS 1.2.0 release. In this version, we provide the code for [Moirai-MoE](https://arxiv.org/abs/2410.10469) which is our latest model. The model weights can be downloaded from huggingface: [small](https://huggingface.co/Salesforce/moirai-moe-1.0-R-small) and [base](https://huggingface.co/Salesforce/moirai-moe-1.0-R-base).
Besides Moirai-MoE, we also provide more evaluation files to facilite the evaluation for some foundation models (e.g., Chronos, TimesFM, VisionTS). Meanwhile, of course, some enhancements and fixes (e.g., save huggingface checkpoint, training resume) are contained in this version :)
What's Changed in the Codebase
Moirai-MoE
* update for moirai-moe by liuxu77 in https://github.com/SalesforceAIResearch/uni2ts/pull/139
* Update readme file by liuxu77 in https://github.com/SalesforceAIResearch/uni2ts/pull/140
* update readme and notebooks for moe by liuxu77 in https://github.com/SalesforceAIResearch/uni2ts/pull/142
More Evaluation files
* Evaluate chronos on monash by liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/72
* update evaluation code and scripts by liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/113
* Scripts and code to evaluate timesfm by liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/119
* add VisionTS scripts and code by Keytoyze in https://github.com/SalesforceAIResearch/uni2ts/pull/120
* fix VisionTS context len issue by Keytoyze in https://github.com/SalesforceAIResearch/uni2ts/pull/124
Enhancements and Fixes
* Add ckpt_path config to resume training by gorold in https://github.com/SalesforceAIResearch/uni2ts/pull/67
* Documentation by gorold in https://github.com/SalesforceAIResearch/uni2ts/pull/71
* Change to python3 for black in pre-commit by liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/73
* add config file for lightning checkpoint by gorold in https://github.com/SalesforceAIResearch/uni2ts/pull/96
* change the version name and the project name for pypi by liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/106
* Implement a new callback HugginfaceCheckpoint by liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/109
* Manually set 'freq' for specific datasets by zqiao11 in https://github.com/SalesforceAIResearch/uni2ts/pull/110
* Update README and config files with Moirai-1.1-R and fix Moirai-1.0-R model weights by gorold in https://github.com/SalesforceAIResearch/uni2ts/pull/111
* update ipynb examples with 1.1 models by liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/112
* update readme by liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/131
* update README by liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/152
* update version by liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/153
* Update __init__.py adding version by liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/155
* update workflows by liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/157, https://github.com/SalesforceAIResearch/uni2ts/pull/158
New Contributors
* liu-jc made their first contribution in https://github.com/SalesforceAIResearch/uni2ts/pull/72
* zqiao11 made their first contribution in https://github.com/SalesforceAIResearch/uni2ts/pull/110
* Keytoyze made their first contribution in https://github.com/SalesforceAIResearch/uni2ts/pull/120
* liuxu77 made their first contribution in https://github.com/SalesforceAIResearch/uni2ts/pull/139
**Full Changelog**: https://github.com/SalesforceAIResearch/uni2ts/compare/1.1.0...1.2.0