Duplicate from Salesforce/moirai-1.1-R-base

Co-authored-by: Doyen <[email protected]>

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,9 @@
+---
+tags:
+- pytorch_model_hub_mixin
+- model_hub_mixin
+---
+
+This is new updated version of Moirai-1.0-R (https://huggingface.co/Salesforce/moirai-1.0-R-base).
+The new Moirai model achieved significant improvements (~20%) for low-frequency cases like Yearly and Quarterly data in Normalised Mean Absolute Error (NMAE) for 40 datasets on the Monash repository.
+

config.json

@@ -0,0 +1,33 @@
+{
+  "attn_dropout_p": 0.0,
+  "d_model": 768,
+  "distr_output": {
+    "_target_": "uni2ts.distribution.mixture.MixtureOutput",
+    "components": [
+      {
+        "_target_": "uni2ts.distribution.student_t.StudentTOutput"
+      },
+      {
+        "_target_": "uni2ts.distribution.normal.NormalFixedScaleOutput",
+        "scale": 0.001
+      },
+      {
+        "_target_": "uni2ts.distribution.negative_binomial.NegativeBinomialOutput"
+      },
+      {
+        "_target_": "uni2ts.distribution.log_normal.LogNormalOutput"
+      }
+    ]
+  },
+  "dropout_p": 0.0,
+  "max_seq_len": 512,
+  "num_layers": 12,
+  "patch_sizes": [
+    8,
+    16,
+    32,
+    64,
+    128
+  ],
+  "scaling": true
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a533dc23e2981c558ec030d49260053b9e93d87d3deb8bf22cb158118ba29b0e
+size 365449072