upload

model_index.json

@@ -0,0 +1,12 @@
+{
+  "_class_name": "DanceDiffusionPipeline",
+  "_diffusers_version": "0.5.0.dev0",
+  "scheduler": [
+    "diffusers",
+    "IPNDMScheduler"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet1DModel"
+  ]
+}

scheduler/scheduler_config.json

@@ -0,0 +1,5 @@
+{
+  "_class_name": "IPNDMScheduler",
+  "_diffusers_version": "0.5.0.dev0",
+  "num_train_timesteps": 100
+}

config.json → unet/config.json

@@ -34,8 +34,11 @@
   "flip_sin_to_cos": true,
   "freq_shift": 0,
   "in_channels": 2,
+  "mid_block_type": "UNetMidBlock1D",
+  "out_block_type": "",
   "out_channels": 2,
-  "sample_size": null,
+  "sample_rate": 16000,
+  "sample_size": 65536,
   "time_embedding_type": "fourier",
   "up_block_types": [
     "AttnUpBlock1D",

diffusion_pytorch_model.bin → unet/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:372bdc6b8fd460610e52d3b0f68540867f0d12b3dd80b6e7b2e3b421b0d108d5
+oid sha256:2f301200659cfb31aa5192e407cab71759d8382aba5f7406fe7d31db90c445aa
 size 885945253