From ab72f21eff07c57a547076c2e3aacf0d58a28343 Mon Sep 17 00:00:00 2001 From: liuyebo <1515783401@qq.com> Date: Mon, 1 Jul 2024 11:10:06 +0800 Subject: [PATCH] =?UTF-8?q?=E5=B0=9D=E8=AF=95=E4=BD=BF=E7=94=A8cpu?= =?UTF-8?q?=E8=BF=9B=E8=A1=8C=E8=AF=86=E5=88=AB?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- photo_review/photo_review.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/photo_review/photo_review.py b/photo_review/photo_review.py index bec0b21..c00fc0a 100644 --- a/photo_review/photo_review.py +++ b/photo_review/photo_review.py @@ -395,11 +395,12 @@ def main(): if phhds: ie_task_flows = [ Taskflow("information_extraction", schema=SETTLEMENT_LIST_SCHEMA, model="uie-x-base", - task_path="config/model/settlement_list_model", layout_analysis=LAYOUT_ANALYSIS), + task_path="config/model/settlement_list_model", layout_analysis=LAYOUT_ANALYSIS, device_id=-1), Taskflow("information_extraction", schema=DISCHARGE_RECORD_SCHEMA, model="uie-x-base", - task_path="config/model/discharge_record_model", layout_analysis=LAYOUT_ANALYSIS), + task_path="config/model/discharge_record_model", layout_analysis=LAYOUT_ANALYSIS, + device_id=-1), Taskflow("information_extraction", schema=COST_LIST_SCHEMA, model="uie-x-base", - task_path="config/model/cost_list_model", layout_analysis=LAYOUT_ANALYSIS), + task_path="config/model/cost_list_model", layout_analysis=LAYOUT_ANALYSIS, device_id=-1), ] for phhd in phhds: pk_phhd = phhd.pk_phhd