diff --git a/examples/int8_training/Finetune_flan_t5_large_bnb_peft.ipynb b/examples/int8_training/Finetune_flan_t5_large_bnb_peft.ipynb
index a12209d..75c41d6 100644
--- a/examples/int8_training/Finetune_flan_t5_large_bnb_peft.ipynb
+++ b/examples/int8_training/Finetune_flan_t5_large_bnb_peft.ipynb
@@ -69,7 +69,7 @@
       ],
       "source": [
         "!pip install -q bitsandbytes datasets accelerate loralib\n",
-        "!pip install -q git+https://github.com/huggingface/transformers.git@main git+https://github.com/younesbelkada/peft.git@int8-wrapper"
+        "!pip install -q git+https://github.com/huggingface/transformers.git@main git+https://github.com/huggingface/peft.git@main"
       ]
     },
     {
@@ -761,6 +761,7 @@
         "text_column = \"sentence\"\n",
         "label_column = \"text_label\"\n",
         "max_length=128\n",
+        "\n",
         "def preprocess_function(examples):\n",
         "    inputs = examples[text_column]\n",
         "    targets = examples[label_column]\n",
@@ -1285,7 +1286,7 @@
       "name": "python",
       "nbconvert_exporter": "python",
       "pygments_lexer": "ipython3",
-      "version": "3.9.16"
+      "version": "3.8.16"
     },
     "vscode": {
       "interpreter": {