From 80e5807bf6d6c6c5d5d9f9ab55725da322efabef Mon Sep 17 00:00:00 2001 From: leiqichn Date: Fri, 30 Aug 2024 23:19:36 +0800 Subject: [PATCH] feat: add finetuning dependency packages --- finetune_demo/requirements.txt | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/finetune_demo/requirements.txt b/finetune_demo/requirements.txt index 702c04c..e8d4e73 100644 --- 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 \ No newline at end of file +ruamel.yaml==0.18.6 +tiktoken>=0.7.0 +typer>=0.12.3 \ No newline at end of file