From 82d5f29842d067ea4e70907f4824c570cdfc6f94 Mon Sep 17 00:00:00 2001
From: YaoyaoChang <cyy574006791@qq.com>
Date: Thu, 4 Dec 2025 07:20:36 -0800
Subject: [PATCH] Fix: Colab downloads occasionally get stuck

---
 demo/vibevoice_realtime_colab.ipynb | 33 ++++++++++++++++++++++++++++-
 1 file changed, 32 insertions(+), 1 deletion(-)

diff --git a/demo/vibevoice_realtime_colab.ipynb b/demo/vibevoice_realtime_colab.ipynb
index dc5c7dc..576de74 100644
--- a/demo/vibevoice_realtime_colab.ipynb
+++ b/demo/vibevoice_realtime_colab.ipynb
@@ -71,10 +71,41 @@
         "print(\"✅ Installed dependencies\")\n",
         "\n",
         "# Download model\n",
-        "!HF_XET_HIGH_PERFORMANCE=1 hf download microsoft/VibeVoice-Realtime-0.5B --quiet  --local-dir /content/models/VibeVoice-Realtime-0.5B > /dev/null\n",
+        "from huggingface_hub import snapshot_download\n",
+        "snapshot_download(\"microsoft/VibeVoice-Realtime-0.5B\", local_dir=\"/content/models/VibeVoice-Realtime-0.5B\")\n",
         "print(\"✅ Downloaded model: microsoft/VibeVoice-Realtime-0.5B\")\n"
       ]
     },
+    {
+      "cell_type": "markdown",
+      "id": "88c727ab",
+      "metadata": {},
+      "source": [
+        "[Optional] If the download exceeds 1 minute, it is probably stuck. You can: (1) interrupt the execution, (2) log in to Hugging Face, and (3) try download again."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "id": "dec6b870",
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "from huggingface_hub import login\n",
+        "login()"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "id": "c579654b",
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "snapshot_download(\"microsoft/VibeVoice-Realtime-0.5B\", local_dir=\"/content/models/VibeVoice-Realtime-0.5B\")\n",
+        "print(\"✅ Downloaded model: microsoft/VibeVoice-Realtime-0.5B\")"
+      ]
+    },
     {
       "cell_type": "markdown",
       "id": "pgKlV7153Ifi",