update readme (#131)

SalesforceAIResearch · Oct 24, 2024 · 959622c · 959622c
1 parent 2ba614d
commit 959622c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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",