refactor(chinese-instruct): 修改 process_func 函数返回值格式
This commit is contained in:
+1
-1
@@ -12,7 +12,7 @@ def build_prompt(query, response) -> str:
|
|||||||
def process_func(input_dict: dict):
|
def process_func(input_dict: dict):
|
||||||
query = input_dict["prompt"]
|
query = input_dict["prompt"]
|
||||||
response = input_dict["response"]
|
response = input_dict["response"]
|
||||||
return build_prompt(query, response)
|
return {"text": build_prompt(query, response)}
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|||||||
Reference in New Issue
Block a user