add requirement of finetune

2024-06-19 01:12:20 +08:00 · 2024-06-19 01:12:20 +08:00 · ab011a344e
parent 7c62112619
commit ab011a344e
2 changed files with 4 additions and 2 deletions
--- a/basic_demo/openai_api_server.py
+++ b/basic_demo/openai_api_server.py
@ -497,7 +497,7 @@ async def predict_stream(model_id, gen_params):
            function_call = {"name": None, "arguments": delta_text}
            tool_call = ChatCompletionMessageToolCall(
                index=0,
-                id=generate_id('call_', 24),
+                id=None,
                function=FunctionCall(**function_call),
                type="function"
            )
--- a/finetune_demo/requirements.txt
+++ b/finetune_demo/requirements.txt
@ -2,4 +2,6 @@ jieba>=0.42.1
 datasets>2.20.0
 peft>=0.11.1
 deepspeed>=0.14.3
-nltk==3.8.1
+nltk==3.8.1
+rouge_chinese>=1.0.3
+ruamel.yaml>=0.18.6