push with PyTorchMixin

config.json

@@ -0,0 +1,33 @@
+{
+  "attn_dropout_p": 0.0,
+  "d_model": 384,
+  "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": 6,
+  "patch_sizes": [
+    8,
+    16,
+    32,
+    64,
+    128
+  ],
+  "scaling": true
+}

model.ckpt → model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2c68fbac0e248a378b5cdd8f8fab330db114a6517a534d079abcfb9c29b1ba93
-size 55346458
+oid sha256:84317fe1a53632f2b72574b072f200811f6a98fecefe30244d5e34492c9d7a7c
+size 55320200