From 2dc29186355db5565c05ee2a08d18e7a28697e03 Mon Sep 17 00:00:00 2001 From: mobicham <37179323+mobicham@users.noreply.github.com> Date: Thu, 26 Oct 2023 13:39:24 +0200 Subject: [PATCH] Update README.md --- code/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/code/README.md b/code/README.md index faf0208..1542801 100644 --- a/code/README.md +++ b/code/README.md @@ -6,7 +6,7 @@ First, install the dependencies: pip install transformers[torch] datasets xformers accelerate evaluate trl ``` -Then run the training script ```train_llama2_lowrank_lora.py``` with one of the datasets. +Then set your HuggingFace token ```hf_auth``` and run the training script ```train_llama2_lowrank_lora.py``` with one of the datasets. If you want to use the model for ~25-30% faster prediction, you can merge the weights via the following command: ```Python