push with PyTorchMixin

config.json

@@ -0,0 +1,33 @@
+{
+  "attn_dropout_p": 0.0,
+  "d_model": 1024,
+  "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": 24,
+  "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:91d5f62f6d83996e9653f162ffdfe5adc17e88f3469dae076b3706903e714fa9
-size 1244012940
+oid sha256:02dc158bfa93e4aa1e9db07e333d1f33b57c24fd1f249a1078d483ff016e9696
+size 1243917120