push with PyTorchMixin

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.ckpt → model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2d365787ddbb5d75dd4add79dfaf2321e32902b909c726f739f2e708060a79b5
-size 365498364
+oid sha256:4a524566c83b9507e06b4a6f61ed2de6d5bdb0a9167a73a504b723bcaca3de59
+size 365449072