diff --git a/tts_pytorch.ipynb b/tts_pytorch.ipynb
index 7c722da..d247b2f 100644
--- a/tts_pytorch.ipynb
+++ b/tts_pytorch.ipynb
@@ -8,6 +8,7 @@
    "outputs": [],
    "source": [
     "# Fast install, might break in the future.\n",
+    "!pip install 'safetensors<0.6'\n",
     "!pip install 'sphn<0.2'\n",
     "!pip install --no-deps \"moshi==0.2.11\"\n",
     "# Slow install (will download torch and cuda), but future proof.\n",