some changes

tts_pytorch.ipynb

@@ -7,7 +7,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install \"moshi==0.2.7\""
+    "!pip install \"moshi==0.2.7\"\n",
+    "# If running from a Google Colab, you can instead use the following two lines to prevent PyTorch from being reinstalled.\n",
+    "# !pip install 'sphn < 0.2'# If\n",
+    "# !pip install --no-deps \"moshi==0.2.7"
    ]
   },
   {
@@ -21,7 +24,6 @@
     "import sys\n",
     "\n",
     "import numpy as np\n",
-    "import sphn\n",
     "import torch\n",
     "from moshi.models.loaders import CheckpointInfo\n",
     "from moshi.models.tts import DEFAULT_DSM_TTS_REPO, DEFAULT_DSM_TTS_VOICE_REPO, TTSModel\n",
@@ -76,6 +78,7 @@
     "print(\"Generating audio...\")\n",
     "\n",
     "pcms = []\n",
+    "@torch.no_grad()\n",
     "def _on_frame(frame):\n",
     "    print(\"Step\", len(pcms), end=\"\\r\")\n",
     "    if (frame != -1).all():\n",