valhalla commited on
Commit
0574c49
·
1 Parent(s): 2e75fa0
diffwave/config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DiffWave",
3
+ "_diffusers_version": "0.0.3",
4
+ "diffusion_step_embed_dim_in": 128,
5
+ "diffusion_step_embed_dim_mid": 512,
6
+ "diffusion_step_embed_dim_out": 512,
7
+ "dilation_cycle": 10,
8
+ "in_channels": 1,
9
+ "num_res_layers": 30,
10
+ "out_channels": 1,
11
+ "res_channels": 128,
12
+ "skip_channels": 128
13
+ }
diffwave/diffusion_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e11bb965dc16fe0ee13dbf5cd9762980664ddeb7431604a173df349f6c19594
3
+ size 27891769
model_index.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "BDDM",
3
+ "_diffusers_version": "0.0.3",
4
+ "_module": "pipeline_bddm",
5
+ "diffwave": [
6
+ "pipeline_bddm",
7
+ "DiffWave"
8
+ ],
9
+ "noise_scheduler": [
10
+ "diffusers",
11
+ "DDPMScheduler"
12
+ ]
13
+ }
noise_scheduler/scheduler_config.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DDPMScheduler",
3
+ "_diffusers_version": "0.0.3",
4
+ "beta_end": 0.02,
5
+ "beta_schedule": "linear",
6
+ "beta_start": 0.0001,
7
+ "clip_predicted_image": false,
8
+ "timestep_values": [
9
+ 1.1816375255584717,
10
+ 3.153350830078125,
11
+ 5.641834259033203,
12
+ 9.051384925842285,
13
+ 13.811037063598633,
14
+ 20.59091567993164,
15
+ 30.359878540039062,
16
+ 133.14947509765625
17
+ ],
18
+ "timesteps": 8,
19
+ "trained_betas": [
20
+ 0.0003544672508724034,
21
+ 0.0007220751140266657,
22
+ 0.0014719057362526655,
23
+ 0.0030041460413485765,
24
+ 0.006144929211586714,
25
+ 0.012615003623068333,
26
+ 0.026042096316814423,
27
+ 0.5767547488212585
28
+ ],
29
+ "variance_type": "fixed_small"
30
+ }