added DDIM model

notebooks/test_model.ipynb

@@ -74,7 +74,9 @@
     "\n",
     "#@markdown teticio/audio-diffusion-instrumental-hiphop-256 - trained on instrumental hiphop\n",
     "\n",
-    "model_id = \"teticio/audio-diffusion-256\"  #@param [\"teticio/audio-diffusion-256\", \"teticio/audio-diffusion-breaks-256\", \"audio-diffusion-instrumenal-hiphop-256\"]"
+    "#@markdown teticio/audio-diffusion-ddim-256                - DDIM model trained on my Spotify \"liked\" playlist\n",
+    "\n",
+    "model_id = \"teticio/audio-diffusion-256\"  #@param [\"teticio/audio-diffusion-256\", \"teticio/audio-diffusion-breaks-256\", \"audio-diffusion-instrumenal-hiphop-256\", \"teticio/audio-diffusion-ddim-256\"]"
    ]
   },
   {
@@ -297,7 +299,7 @@
     "    audio2) = audio_diffusion.generate_spectrogram_and_audio_from_audio(\n",
     "        raw_audio=mel.get_audio_slice(slice),\n",
     "        mask_start_secs=1,\n",
-    "        mask_end_secs=1)\n",
+    "        mask_end_secs=1, step_generator=torch.Generator())\n",
     "display(Audio(audio, rate=sample_rate))\n",
     "display(Audio(audio2, rate=sample_rate))"
    ]