Fix not enough resource

aigc-apps · Dec 9, 2024 · 1aeaf63 · 1aeaf63
1 parent 3fb4d33
commit 1aeaf63
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git 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
@@ -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,
     )