doyensahoo gorold commited on
Commit
c4f3b41
1 Parent(s): 001acc2

push with PyTorchMixin (#1)

Browse files

- push with PyTorchMixin (a93f42656a2377e3417d03df677258d0a80d3a38)


Co-authored-by: Gerald Woo <[email protected]>

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": 1024,
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": 24,
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:91d5f62f6d83996e9653f162ffdfe5adc17e88f3469dae076b3706903e714fa9
3
- size 1244012940
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:02dc158bfa93e4aa1e9db07e333d1f33b57c24fd1f249a1078d483ff016e9696
3
+ size 1243917120