diff --git "a/notebooks/test_model.ipynb" "b/notebooks/test_model.ipynb"
--- "a/notebooks/test_model.ipynb"
+++ "b/notebooks/test_model.ipynb"
@@ -128,7 +128,7 @@
"id": "80e5b5fa",
"metadata": {},
"source": [
- "Try playing around with `start_steps`. Values closer to zero will produce new samples, while values closer to `steps` will produce samples more faithful to the original."
+ "Try playing around with `start_steps`. Values closer to zero will produce new samples, while values closer to 1,000 will produce samples more faithful to the original."
]
},
{
@@ -162,8 +162,7 @@
" sample_rate, audio2\n",
" ) = audio_diffusion.generate_spectrogram_and_audio_from_audio(\n",
" raw_audio=audio,\n",
- " start_step=start_steps,\n",
- " steps=1000)\n",
+ " start_step=start_steps)\n",
" display(image2)\n",
" display(Audio(audio2, rate=sample_rate))\n",
" track = np.concatenate([track, AudioDiffusion.loop_it(audio2, sample_rate, loops=1)])\n",
@@ -193,125 +192,14 @@
"metadata": {
"scrolled": true
},
- "outputs": [
- {
- "data": {
- "application/vnd.jupyter.widget-view+json": {
- "model_id": "06e31541e8284faa9c71dafb5b686574",
- "version_major": 2,
- "version_minor": 0
- },
- "text/plain": [
- " 0%| | 0/500 [00:00, ?it/s]"
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- },
- {
- "data": {
- "text/html": [
- "\n",
- " \n",
- " "
- ],
- "text/plain": [
- ""
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- },
- {
- "data": {
- "text/html": [
- "\n",
- " \n",
- " "
- ],
- "text/plain": [
- ""
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- },
- {
- "data": {
- "application/vnd.jupyter.widget-view+json": {
- "model_id": "b9e0e92b62404876b1600c7f01408503",
- "version_major": 2,
- "version_minor": 0
- },
- "text/plain": [
- " 0%| | 0/500 [00:00, ?it/s]"
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- },
- {
- "data": {
- "text/html": [
- "\n",
- " \n",
- " "
- ],
- "text/plain": [
- ""
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- },
- {
- "data": {
- "text/html": [
- "\n",
- " \n",
- " "
- ],
- "text/plain": [
- ""
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- },
- {
- "data": {
- "application/vnd.jupyter.widget-view+json": {
- "model_id": "116c3bc7b4c2449fa220e93088a26d6f",
- "version_major": 2,
- "version_minor": 0
- },
- "text/plain": [
- " 0%| | 0/500 [00:00, ?it/s]"
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- }
- ],
+ "outputs": [],
"source": [
"start_steps = 500 #@param {type:\"slider\", min:0, max:1000, step:10}\n",
"audio_file = \"/home/teticio/Music/Music/Cesar Mariano And CIA/Gilles Peterson In Brazil_ Da Hora/2-07 Futebol De Bar (Heavy Ãœsker Mix.mp3\" #@param {type:\"string\"}\n",
"audio_diffusion.mel.load_audio(audio_file)\n",
"track = np.array([])\n",
- "generator = torch.Generator().manual_seed(seed)\n",
- "seed = 16183389798189209330 #generator.seed()\n",
+ "generator = torch.Generator()\n",
+ "seed = generator.seed()\n",
"for slice in range(audio_diffusion.mel.get_number_of_slices()):\n",
" generator.manual_seed(seed)\n",
" audio = audio_diffusion.mel.get_audio_slice(slice)\n",
@@ -321,7 +209,6 @@
" audio_file=audio_file,\n",
" slice=slice,\n",
" start_step=start_steps,\n",
- " steps=1000,\n",
" generator=generator)\n",
" display(Audio(audio, rate=sample_rate))\n",
" display(Audio(audio2, rate=sample_rate))\n",
@@ -331,28 +218,10 @@
},
{
"cell_type": "code",
- "execution_count": 39,
+ "execution_count": null,
"id": "90457786",
"metadata": {},
- "outputs": [
- {
- "data": {
- "text/html": [
- "\n",
- " \n",
- " "
- ],
- "text/plain": [
- ""
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- }
- ],
+ "outputs": [],
"source": [
"display(Audio(track, rate=sample_rate))"
]