diff --git a/README.md b/README.md index d34e1c7..f7f986c 100644 --- a/README.md +++ b/README.md @@ -7,6 +7,8 @@ Uni2TS also provides tools for fine-tuning, inference, and evaluation for time s ## 🎉 What's New +* Oct 2024: A new model Moirai-MoE! The preprint is now available on [arXiv](https://arxiv.org/abs/2410.10469). Model weights to be released soon. + * Jun 2024: Released Moirai-1.1-R model weights in [small](https://huggingface.co/Salesforce/moirai-1.1-R-small), [base](https://huggingface.co/Salesforce/moirai-1.1-R-base), and [large](https://huggingface.co/Salesforce/moirai-1.1-R-large). * May 2024: The Uni2TS paper has been accepted to ICML 2024 as an Oral presentation! diff --git a/pyproject.toml b/pyproject.toml index c651e71..f2430af 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -8,7 +8,7 @@ dynamic = [ "version", ] dependencies = [ - "torch>=2.1", + "torch>=2.1,<2.5", "lightning>=2.0", "gluonts~=0.14.3", "numpy~=1.26.0",