diff --git a/sovits4_for_colab.ipynb b/sovits4_for_colab.ipynb
index 394bfd4..ee21b16 100644
--- a/sovits4_for_colab.ipynb
+++ b/sovits4_for_colab.ipynb
@@ -176,7 +176,7 @@
     "    !curl -L {G_0_URL} -o logs/44k/G_0.pth\n",
     "\n",
     "if download_pretrained_diffusion_model:\n",
-    "    !curl -L {diffusion_URL} -o logs/44k/diffusion/model_0.pt"
+    "    !curl -L {diff_model_URL} -o logs/44k/diffusion/model_0.pt"
    ]
   },
   {
@@ -460,6 +460,21 @@
     "#@markdown\n",
     "\n",
     "%cd /content/so-vits-svc\n",
+    "\n",
+    "import os\n",
+    "if not os.path.exists(\"./pretrain/nsf_hifigan/model\"):\n",
+    "  !curl -L https://github.com/openvpi/vocoders/releases/download/nsf-hifigan-v1/nsf_hifigan_20221211.zip -o nsf_hifigan_20221211.zip\n",
+    "  !unzip nsf_hifigan_20221211.zip\n",
+    "  !rm -rf pretrain/nsf_hifigan\n",
+    "  !mv -v nsf_hifigan pretrain\n",
+    "\n",
+    "#@markdown Whether to enable tensorboard\n",
+    "tensorboard_on = True  #@param {type:\"boolean\"}\n",
+    "\n",
+    "if tensorboard_on:\n",
+    "  %load_ext tensorboard\n",
+    "  %tensorboard --logdir logs/44k\n",
+    "\n",
     "!python train_diff.py -c configs/diffusion.yaml"
    ]
   },