From 342e3b86a4323ae790be7b45627b974d201a5075 Mon Sep 17 00:00:00 2001 From: Yongliang Shen Date: Mon, 23 Sep 2024 17:19:56 +0800 Subject: [PATCH] update data --- taskbench/inference.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/taskbench/inference.py b/taskbench/inference.py index 55c7986..8a66045 100644 --- a/taskbench/inference.py +++ b/taskbench/inference.py @@ -36,6 +36,9 @@ def __init__(self, message): @click.option("--log_first_detail", type=bool, default=False) def main(data_dir, temperature, top_p, api_addr, api_key, api_port, multiworker, llm, use_demos, reformat, reformat_by, tag, dependency_type, log_first_detail): assert dependency_type in ["resource", "temporal"], "Dependency type not supported" + if dependency_type == "resource": + assert data_dir != "data_dailylifeapis", "Resource dependency type only support data_huggingface and data_multimedia" + arguments = locals() url = f"http://{api_addr}:{api_port}/v1/chat/completions" header = {