From afa77391316b5a8e0b0cb3729ab345fed49c2c96 Mon Sep 17 00:00:00 2001 From: younesbelkada Date: Wed, 15 Feb 2023 08:01:56 +0000 Subject: [PATCH] update --- examples/int8_training/Finetune_flan_t5_large_bnb_peft.ipynb | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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": {