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