patrickvonplaten commited on
Commit
60fdab4
•
1 Parent(s): 971ba2d
model_index.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DanceDiffusionPipeline",
3
+ "_diffusers_version": "0.5.0.dev0",
4
+ "scheduler": [
5
+ "diffusers",
6
+ "IPNDMScheduler"
7
+ ],
8
+ "unet": [
9
+ "diffusers",
10
+ "UNet1DModel"
11
+ ]
12
+ }
scheduler/scheduler_config.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "IPNDMScheduler",
3
+ "_diffusers_version": "0.5.0.dev0",
4
+ "num_train_timesteps": 100
5
+ }
config.json → unet/config.json RENAMED
@@ -34,8 +34,11 @@
34
  "flip_sin_to_cos": true,
35
  "freq_shift": 0,
36
  "in_channels": 2,
 
 
37
  "out_channels": 2,
38
- "sample_size": null,
 
39
  "time_embedding_type": "fourier",
40
  "up_block_types": [
41
  "AttnUpBlock1D",
 
34
  "flip_sin_to_cos": true,
35
  "freq_shift": 0,
36
  "in_channels": 2,
37
+ "mid_block_type": "UNetMidBlock1D",
38
+ "out_block_type": "",
39
  "out_channels": 2,
40
+ "sample_rate": 48000,
41
+ "sample_size": 65536,
42
  "time_embedding_type": "fourier",
43
  "up_block_types": [
44
  "AttnUpBlock1D",
diffusion_pytorch_model.bin → unet/diffusion_pytorch_model.bin RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:741543bd829c9912c1d68d60769a8183049dc7aedb9dfb0c5d329584b9f994af
3
  size 885945253
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6636c8e3cdb10313eb6bdf868394b3c8ea1ff8fd627439ca1f0e4d75938dac61
3
  size 885945253