diff --git a/HeartLib.ipynb b/HeartLib.ipynb
index 047a847..0126728 100644
--- a/HeartLib.ipynb
+++ b/HeartLib.ipynb
@@ -1,28 +1,10 @@
 {
-  "nbformat": 4,
-  "nbformat_minor": 0,
-  "metadata": {
-    "colab": {
-      "provenance": [],
-      "machine_shape": "hm",
-      "gpuType": "A100",
-      "include_colab_link": true
-    },
-    "kernelspec": {
-      "name": "python3",
-      "display_name": "Python 3"
-    },
-    "language_info": {
-      "name": "python"
-    },
-    "accelerator": "GPU"
-  },
   "cells": [
     {
       "cell_type": "markdown",
       "metadata": {
-        "id": "view-in-github",
-        "colab_type": "text"
+        "colab_type": "text",
+        "id": "view-in-github"
       },
       "source": [
         "<a href=\"https://colab.research.google.com/github/theelderemo/HeartLib-Google-Colab/blob/main/HeartLib.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
@@ -32,8 +14,8 @@
       "cell_type": "code",
       "execution_count": null,
       "metadata": {
-        "id": "YkaqrXnP0WXF",
-        "collapsed": true
+        "collapsed": true,
+        "id": "YkaqrXnP0WXF"
       },
       "outputs": [],
       "source": [
@@ -55,6 +37,11 @@
     },
     {
       "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "OcdLjfex0l6N"
+      },
+      "outputs": [],
       "source": [
         "# @title Download Checkpoints\n",
         "import os\n",
@@ -65,18 +52,18 @@
         "os.makedirs('./ckpt', exist_ok=True)\n",
         "\n",
         "!huggingface-cli download --local-dir './ckpt' 'HeartMuLa/HeartMuLaGen'\n",
-        "!huggingface-cli download --local-dir './ckpt/HeartMuLa-oss-3B' 'HeartMuLa/HeartMuLa-oss-3B'\n",
-        "!huggingface-cli download --local-dir './ckpt/HeartCodec-oss' 'HeartMuLa/HeartCodec-oss'\n",
+        "!huggingface-cli download --local-dir './ckpt/HeartMuLa-oss-3B' 'HeartMuLa/HeartMuLa-RL-oss-3B-20260123'\n",
+        "!huggingface-cli download --local-dir './ckpt/HeartCodec-oss' 'HeartMuLa/HeartCodec-oss-20260123'\n",
         "!huggingface-cli download --local-dir './ckpt/HeartTranscriptor-oss' 'HeartMuLa/HeartTranscriptor-oss'"
-      ],
-      "metadata": {
-        "id": "OcdLjfex0l6N"
-      },
-      "execution_count": null,
-      "outputs": []
+      ]
     },
     {
       "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "m7i-Cm0j0o3d"
+      },
+      "outputs": [],
       "source": [
         "# @title Run Music Generation\n",
         "# No Touchy\n",
@@ -128,12 +115,25 @@
         "    display(Audio(\"./assets/output.mp3\"))\n",
         "else:\n",
         "    print(\"✗ Generation failed. Check logs above.\")\n"
-      ],
-      "metadata": {
-        "id": "m7i-Cm0j0o3d"
-      },
-      "execution_count": null,
-      "outputs": []
+      ]
+    }
+  ],
+  "metadata": {
+    "accelerator": "GPU",
+    "colab": {
+      "gpuType": "A100",
+      "include_colab_link": true,
+      "machine_shape": "hm",
+      "provenance": []
+    },
+    "kernelspec": {
+      "display_name": "Python 3",
+      "name": "python3"
+    },
+    "language_info": {
+      "name": "python"
     }
-  ]
-}
\ No newline at end of file
+  },
+  "nbformat": 4,
+  "nbformat_minor": 0
+}
diff --git a/heartlib.py b/heartlib.py
index 542ceb2..d4c82c7 100644
--- a/heartlib.py
+++ b/heartlib.py
@@ -33,8 +33,8 @@
 os.makedirs('./ckpt', exist_ok=True)
 
 !huggingface-cli download --local-dir './ckpt' 'HeartMuLa/HeartMuLaGen'
-!huggingface-cli download --local-dir './ckpt/HeartMuLa-oss-3B' 'HeartMuLa/HeartMuLa-oss-3B'
-!huggingface-cli download --local-dir './ckpt/HeartCodec-oss' 'HeartMuLa/HeartCodec-oss'
+!huggingface-cli download --local-dir './ckpt/HeartMuLa-oss-3B' 'HeartMuLa/HeartMuLa-RL-oss-3B-20260123'
+!huggingface-cli download --local-dir './ckpt/HeartCodec-oss' 'HeartMuLa/HeartCodec-oss-20260123'
 !huggingface-cli download --local-dir './ckpt/HeartTranscriptor-oss' 'HeartMuLa/HeartTranscriptor-oss'
 
 # Commented out IPython magic to ensure Python compatibility.