From 101d314bd9437a236e485956438ef48a867aea78 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 27 Mar 2023 16:13:46 -0400 Subject: [PATCH] add missing dependency to train with LlamaTokenizer --- requirements.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/requirements.txt b/requirements.txt index dadcad1..c752c4b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,6 +2,7 @@ torch accelerate bitsandbytes datasets +sentencepiece git+https://github.com/huggingface/transformers.git git+https://github.com/sterlind/GPTQ-for-LLaMa.git@lora_4bit git+https://github.com/sterlind/peft.git