push with PyTorchMixin

#2
Files changed (2) hide show
  1. config.json +33 -0
  2. model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "attn_dropout_p": 0.0,
3
+ "d_model": 384,
4
+ "distr_output": {
5
+ "_target_": "uni2ts.distribution.mixture.MixtureOutput",
6
+ "components": [
7
+ {
8
+ "_target_": "uni2ts.distribution.student_t.StudentTOutput"
9
+ },
10
+ {
11
+ "_target_": "uni2ts.distribution.normal.NormalFixedScaleOutput",
12
+ "scale": 0.001
13
+ },
14
+ {
15
+ "_target_": "uni2ts.distribution.negative_binomial.NegativeBinomialOutput"
16
+ },
17
+ {
18
+ "_target_": "uni2ts.distribution.log_normal.LogNormalOutput"
19
+ }
20
+ ]
21
+ },
22
+ "dropout_p": 0.0,
23
+ "max_seq_len": 512,
24
+ "num_layers": 6,
25
+ "patch_sizes": [
26
+ 8,
27
+ 16,
28
+ 32,
29
+ 64,
30
+ 128
31
+ ],
32
+ "scaling": true
33
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:84317fe1a53632f2b72574b072f200811f6a98fecefe30244d5e34492c9d7a7c
3
+ size 55320200