push with PyTorchMixin

#3
Files changed (2) hide show
  1. config.json +33 -0
  2. model.ckpt → model.safetensors +2 -2
config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "attn_dropout_p": 0.0,
3
+ "d_model": 768,
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": 12,
25
+ "patch_sizes": [
26
+ 8,
27
+ 16,
28
+ 32,
29
+ 64,
30
+ 128
31
+ ],
32
+ "scaling": true
33
+ }
model.ckpt → model.safetensors RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2d365787ddbb5d75dd4add79dfaf2321e32902b909c726f739f2e708060a79b5
3
- size 365498364
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4a524566c83b9507e06b4a6f61ed2de6d5bdb0a9167a73a504b723bcaca3de59
3
+ size 365449072