From 1aeaf632e3efb2f8a2cfc0841db6e411c6dbe958 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=AD=B1=E6=96=87?= Date: Mon, 9 Dec 2024 14:29:23 +0800 Subject: [PATCH] Fix not enough resource --- src/pai_rag/tools/data_process/embed_workflow.py | 1 - src/pai_rag/tools/data_process/split_workflow.py | 1 - 2 files changed, 2 deletions(-) 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, )