fix prompt lag w override

yyahav · Oct 9, 2022 · 6b7ceb4 · 6b7ceb4
1 parent 1680cc1
commit 6b7ceb4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Video_Killed_The_Radio_Star_Defusion.ipynb b/Video_Killed_The_Radio_Star_Defusion.ipynb
@@ -795,7 +795,7 @@
         "    \n",
         "    if prompt_lag and (idx > 0):\n",
         "        rec_prev = prompt_starts[idx -1]\n",
-        "        prev_prompt = rec_prev['override_prompt']\n",
+        "        prev_prompt = rec_prev.get('override_prompt','').strip()\n",
         "        if not prev_prompt:\n",
         "            prev_prompt = rec_prev['prompt']\n",
         "        prompt = f\"{prev_prompt}, {prompt}\"\n",
@@ -1062,7 +1062,7 @@
         "try:\n",
         "    from prefetch_generator import BackgroundGenerator\n",
         "except:\n",
-        "    !pip install prefetch_generator \n",
+        "    !pip install prefetch_generator\n",
         "    from prefetch_generator import BackgroundGenerator\n",
         "\n",
         "from vktrs.tsp import (\n",