Duplicate from kashif/music-spectrogram-diffusion

Co-authored-by: Kashif Rasul <[email protected]>

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,30 @@
+---
+license: apache-2.0
+tags:
+- pytorch
+- diffusers
+duplicated_from: kashif/music-spectrogram-diffusion
+---
+
+# Multi-instrument Music Synthesis with Spectrogram Diffusion
+
+[Spectrogram Diffusion](https://arxiv.org/abs/2206.05408) by Curtis Hawthorne, Ian Simon, Adam Roberts, Neil Zeghidour, Josh Gardner, Ethan Manilow, and Jesse Engel.
+
+## Abstract
+
+An ideal music synthesizer should be both interactive and expressive, generating high-fidelity audio in realtime for arbitrary combinations of instruments and notes. Recent neural synthesizers have exhibited a tradeoff between domain-specific models that offer detailed control of only specific instruments, or raw waveform models that can train on any music but with minimal control and slow generation. In this work, we focus on a middle ground of neural synthesizers that can generate audio from MIDI sequences with arbitrary combinations of instruments in realtime. This enables training on a wide range of transcription datasets with a single model, which in turn offers note-level control of composition and instrumentation across a wide range of instruments. We use a simple two-stage process: MIDI to spectrograms with an encoder-decoder Transformer, then spectrograms to audio with a generative adversarial network (GAN) spectrogram inverter. We compare training the decoder as an autoregressive model and as a Denoising Diffusion Probabilistic Model (DDPM) and find that the DDPM approach is superior both qualitatively and as measured by audio reconstruction and Fréchet distance metrics. Given the interactivity and generality of this approach, we find this to be a promising first step towards interactive and expressive neural synthesis for arbitrary combinations of instruments and notes.
+
+<img src="https://storage.googleapis.com/music-synthesis-with-spectrogram-diffusion/architecture.png" alt="Architecture diagram">
+
+## Example usage
+
+```python
+from diffusers import SpectrogramDiffusionPipeline
+
+pipe = SpectrogramDiffusionPipeline.from_pretrained("kashif/music-spectrogram-diffusion")
+pipe = pipe.to("cuda")
+
+output = pipe("beethoven_hammerklavier_2.mid")
+
+audio = output.audios[0]
+```

continuous_encoder/config.json

@@ -0,0 +1,14 @@
+{
+  "_class_name": "SpectrogramContEncoder",
+  "_diffusers_version": "0.14.0.dev0",
+  "d_ff": 2048,
+  "d_kv": 64,
+  "d_model": 768,
+  "dropout_rate": 0.1,
+  "feed_forward_proj": "gated-gelu",
+  "input_dims": 128,
+  "is_decoder": false,
+  "num_heads": 12,
+  "num_layers": 12,
+  "targets_context_length": 256
+}

continuous_encoder/diffusion_pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9ce0037cc2f9aad486141769202fed6a112db29487a358651424d1354ba52038
+size 341044659

decoder/config.json

@@ -0,0 +1,13 @@
+{
+  "_class_name": "T5FilmDecoder",
+  "_diffusers_version": "0.14.0.dev0",
+  "d_ff": 2048,
+  "d_kv": 64,
+  "d_model": 768,
+  "dropout_rate": 0.1,
+  "input_dims": 128,
+  "max_decoder_noise_time": 20000.0,
+  "num_heads": 12,
+  "num_layers": 12,
+  "targets_length": 256
+}

decoder/diffusion_pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eda770548b0ac2432973dee6a99068fdf2120454e7599ea3b62acebde6aa1d5b
+size 954931957

melgan/model.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0f7bfb7c042cfed63b133aa26ec440f7b2d08192823fbc2363499696b3720603
+size 60487709

model_index.json

@@ -0,0 +1,24 @@
+{
+  "_class_name": "SpectrogramDiffusionPipeline",
+  "_diffusers_version": "0.14.0.dev0",
+  "continuous_encoder": [
+    "spectrogram_diffusion",
+    "SpectrogramContEncoder"
+  ],
+  "decoder": [
+    "diffusers",
+    "T5FilmDecoder"
+  ],
+  "melgan": [
+    "diffusers",
+    "OnnxRuntimeModel"
+  ],
+  "notes_encoder": [
+    "spectrogram_diffusion",
+    "SpectrogramNotesEncoder"
+  ],
+  "scheduler": [
+    "diffusers",
+    "DDPMScheduler"
+  ]
+}

notes_encoder/config.json

@@ -0,0 +1,14 @@
+{
+  "_class_name": "SpectrogramNotesEncoder",
+  "_diffusers_version": "0.14.0.dev0",
+  "d_ff": 2048,
+  "d_kv": 64,
+  "d_model": 768,
+  "dropout_rate": 0.1,
+  "feed_forward_proj": "gated-gelu",
+  "is_decoder": false,
+  "max_length": 2048,
+  "num_heads": 12,
+  "num_layers": 12,
+  "vocab_size": 1536
+}

notes_encoder/diffusion_pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:be7f951c6ed13ef10ae9298c10c315d9d27a29a4da42fad9f24355967fd37e4c
+size 350875059

scheduler/scheduler_config.json

@@ -0,0 +1,13 @@
+{
+  "_class_name": "DDPMScheduler",
+  "_diffusers_version": "0.14.0.dev0",
+  "beta_end": 0.02,
+  "beta_schedule": "squaredcos_cap_v2",
+  "beta_start": 0.0001,
+  "clip_sample": true,
+  "clip_sample_range": 1.0,
+  "num_train_timesteps": 1000,
+  "prediction_type": "epsilon",
+  "trained_betas": null,
+  "variance_type": "fixed_large"
+}