updated pipeline

continuous_encoder/config.json

@@ -0,0 +1,14 @@
+{
+  "_class_name": "SpectrogramContEncoder",
+  "_diffusers_version": "0.9.0",
+  "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,14 @@
+{
+  "_class_name": "T5FilmDecoder",
+  "_diffusers_version": "0.9.0",
+  "d_ff": 2048,
+  "d_kv": 64,
+  "d_model": 768,
+  "dropout_rate": 0.1,
+  "feed_forward_proj": "gated-gelu",
+  "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:9e232232517a2dbaf4f345a0fba2394d3a0e70a461fa73e52e7279054f749bff
+size 954929397

model_index.json

@@ -1,9 +1,17 @@
 {
   "_class_name": "SpectrogramDiffusionPipeline",
-  "_diffusers_version": "0.7.0",
-  "cont_context_trans": [
+  "_diffusers_version": "0.9.0",
+  "continuous_encoder": [
     "spectrogram_diffusion",
-    "ContinuousContextTransformer"
+    "SpectrogramContEncoder"
+  ],
+  "decoder": [
+    "spectrogram_diffusion",
+    "T5FilmDecoder"
+  ],
+  "notes_encoder": [
+    "spectrogram_diffusion",
+    "SpectrogramNotesEncoder"
   ],
   "scheduler": [
     "diffusers",

notes_encoder/config.json

@@ -0,0 +1,14 @@
+{
+  "_class_name": "SpectrogramNotesEncoder",
+  "_diffusers_version": "0.9.0",
+  "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