reproducible samples with seed

notebooks/test_model.ipynb

@@ -84,17 +84,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "audio_diffusion = AudioDiffusion(model_id=model_id)"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "4dc17ac0",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "mel = Mel(x_res=256, y_res=256)"
+    "audio_diffusion = AudioDiffusion(model_id=model_id)\n",
+    "mel = Mel(x_res=256, y_res=256)\n",
+    "generator = torch.Generator()"
    ]
   },
   {
@@ -112,10 +104,13 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "generator = torch.Generator()\n",
     "for _ in range(10):\n",
-    "    print(f'Seed = {generator.seed()}')\n",
-    "    image, (sample_rate, audio) = audio_diffusion.generate_spectrogram_and_audio(generator)\n",
+    "    seed = generator.seed()\n",
+    "    print(f'Seed = {seed}')\n",
+    "    generator.manual_seed(seed)\n",
+    "    image, (sample_rate,\n",
+    "            audio) = audio_diffusion.generate_spectrogram_and_audio(\n",
+    "                generator=generator)\n",
     "    display(image)\n",
     "    display(Audio(audio, rate=sample_rate))\n",
     "    loop = AudioDiffusion.loop_it(audio, sample_rate)\n",
@@ -149,9 +144,10 @@
    "outputs": [],
    "source": [
     "seed = 16183389798189209330  #@param {type:\"integer\"}\n",
+    "generator.manual_seed(seed)\n",
     "image, (sample_rate,\n",
-    "        audio) = audio_diffusion.generate_spectrogram_and_audio_from_audio(\n",
-    "            generator=torch.Generator().manual_seed(seed))\n",
+    "        audio) = audio_diffusion.generate_spectrogram_and_audio(\n",
+    "            generator=generator)\n",
     "display(image)\n",
     "display(Audio(audio, rate=sample_rate))"
    ]
@@ -258,7 +254,6 @@
     "overlap_samples = overlap_secs * mel.get_sample_rate()\n",
     "slice_size = mel.x_res * mel.hop_length\n",
     "stride = slice_size - overlap_samples\n",
-    "generator = torch.Generator()\n",
     "seed = generator.seed()\n",
     "print(f'Seed = {seed}')\n",
     "track = np.array([])\n",
@@ -346,6 +341,14 @@
     "audio = mel.image_to_audio(image)\n",
     "Audio(data=audio, rate=mel.get_sample_rate())"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "4deb47f4",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {