diff --git a/docs/tutorials/diffusion_quantization_acceleration.ipynb b/docs/tutorials/diffusion_quantization_acceleration.ipynb
index 8f0c6928..e89ed0e6 100644
--- a/docs/tutorials/diffusion_quantization_acceleration.ipynb
+++ b/docs/tutorials/diffusion_quantization_acceleration.ipynb
@@ -138,19 +138,7 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {},
-   "outputs": [
-    {
-     "ename": "",
-     "evalue": "",
-     "output_type": "error",
-     "traceback": [
-      "\u001b[1;31mThe Kernel crashed while executing code in the current cell or a previous cell. \n",
-      "\u001b[1;31mPlease review the code in the cell(s) to identify a possible cause of the failure. \n",
-      "\u001b[1;31mClick <a href='https://aka.ms/vscodeJupyterKernelCrash'>here</a> for more info. \n",
-      "\u001b[1;31mView Jupyter <a href='command:jupyter.viewOutput'>log</a> for further details."
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "# Define the prompt\n",
     "prompt = \"a smiling cat dancing on a table. Miyazaki style\"\n",
@@ -167,7 +155,7 @@
    ]
   },
   {
-   "cell_type": "raw",
+   "cell_type": "markdown",
    "metadata": {
     "raw_mimetype": "text/restructuredtext",
     "vscode": {
@@ -183,7 +171,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "pruna",
+   "display_name": ".venv",
    "language": "python",
    "name": "python3"
   },
@@ -197,7 +185,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.11"
+   "version": "3.12.11"
   }
  },
  "nbformat": 4,
diff --git a/docs/tutorials/video_generation.ipynb b/docs/tutorials/video_generation.ipynb
index e79f64a4..2face943 100644
--- a/docs/tutorials/video_generation.ipynb
+++ b/docs/tutorials/video_generation.ipynb
@@ -130,20 +130,6 @@
     "export_to_video(output, \"base_video.mp4\", fps=15)"
    ]
   },
-  {
-   "cell_type": "raw",
-   "metadata": {
-    "vscode": {
-     "languageId": "raw"
-    }
-   },
-   "source": [
-    "<video controls width=\"100%\" style=\"max-width:640px; height:auto;\">\n",
-    "    <source src=\"../assets/videos/video_generation/base_video.mp4\" type=\"video/mp4\">\n",
-    "    Your browser does not support HTML5 video.\n",
-    "</video>"
-   ]
-  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -239,20 +225,6 @@
     "export_to_video(output, \"smashed_video.mp4\", fps=15)"
    ]
   },
-  {
-   "cell_type": "raw",
-   "metadata": {
-    "vscode": {
-     "languageId": "raw"
-    }
-   },
-   "source": [
-    "<video controls width=\"100%\" style=\"max-width:640px; height:auto;\">\n",
-    "    <source src=\"../assets/videos/video_generation/smashed_video.mp4\" type=\"video/mp4\">\n",
-    "    Your browser does not support HTML5 video.\n",
-    "</video>"
-   ]
-  },
   {
    "cell_type": "markdown",
    "metadata": {},
diff --git a/docs/user_manual/configure.rst b/docs/user_manual/configure.rst
index 4bfb8a67..be034f7e 100644
--- a/docs/user_manual/configure.rst
+++ b/docs/user_manual/configure.rst
@@ -248,7 +248,7 @@ Underneath you can find the list of all the available datasets.
      - Custom Collate Function
      - Collate Function Arguments
    * - Text Generation
-     - `WikiText <https://huggingface.co/datasets/Salesforce/wikitext>`_, `SmolTalk <https://huggingface.co/datasets/HuggingFaceTB/smoltalk>`_, `SmolSmolTalk <https://huggingface.co/datasets/HuggingFaceTB/smol-smoltalk>`_, `PubChem <https://huggingface.co/datasets/alxfgh/PubChem10M_SELFIES>`_, `OpenAssistant <https://huggingface.co/datasets/timdettmers/openassistant-guanaco>`_, `C4 <https://huggingface.co/datasets/allenai/c4>`_
+     - `WikiText <https://huggingface.co/datasets/Salesforce/wikitext>`_, `TinyWikiText <https://huggingface.co/datasets/mikasenghaas/wikitext-2>`_, `SmolTalk <https://huggingface.co/datasets/HuggingFaceTB/smoltalk>`_, `SmolSmolTalk <https://huggingface.co/datasets/HuggingFaceTB/smol-smoltalk>`_, `PubChem <https://huggingface.co/datasets/alxfgh/PubChem10M_SELFIES>`_, `OpenAssistant <https://huggingface.co/datasets/timdettmers/openassistant-guanaco>`_, `C4 <https://huggingface.co/datasets/allenai/c4>`_, `TinyIMDB <https://huggingface.co/datasets/stanfordnlp/imdb>`_, `C4 <https://huggingface.co/datasets/allenai/c4>`_
      - ``text_generation_collate``
      - ``text: str``
    * - Image Generation