feat: add finetuning dependency packages

2024-08-30 23:19:36 +08:00 · 2024-08-30 23:19:36 +08:00 · 80e5807bf6
parent f0d67ff4a4
commit 80e5807bf6
1 changed files with 3 additions and 1 deletions
--- a/finetune_demo/requirements.txt
+++ b/finetune_demo/requirements.txt
@ -4,4 +4,6 @@ peft==0.12.2
 deepspeed==0.14.4
 nltk==3.8.1
 rouge_chinese==1.0.3
-ruamel.yaml==0.18.6
+ruamel.yaml==0.18.6
+tiktoken>=0.7.0
+typer>=0.12.3