initial lora training tab

2023-03-25 12:08:26 -07:00
parent 8c8e8b4450
commit 566898a79a
7 changed files with 153 additions and 3 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,4 +10,6 @@ rwkv==0.7.0
 safetensors==0.3.0
 sentencepiece
 tqdm
+peft
+datasets
 git+https://github.com/huggingface/transformers