diff --git a/src/pai_rag/tools/data_process/embed_workflow.py b/src/pai_rag/tools/data_process/embed_workflow.py index 331339ce..2694a987 100644 --- a/src/pai_rag/tools/data_process/embed_workflow.py +++ b/src/pai_rag/tools/data_process/embed_workflow.py @@ -17,7 +17,6 @@ def main(args): "working_dir": args.working_dir, "config_file": args.config_file, }, - num_cpus=args.num_cpus, concurrency=get_concurrency(), batch_size=args.batch_size, ) diff --git a/src/pai_rag/tools/data_process/split_workflow.py b/src/pai_rag/tools/data_process/split_workflow.py index cb0380f7..176acf75 100644 --- a/src/pai_rag/tools/data_process/split_workflow.py +++ b/src/pai_rag/tools/data_process/split_workflow.py @@ -18,7 +18,6 @@ def main(args): "working_dir": args.working_dir, "config_file": args.config_file, }, - num_cpus=args.num_cpus, concurrency=get_concurrency(), batch_size=args.batch_size, )