From 3645e13f00c848c0eeea4808df748c106a4d1366 Mon Sep 17 00:00:00 2001 From: liuyebo <1515783401@qq.com> Date: Wed, 26 Jun 2024 13:47:31 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9A=82=E6=97=B6=E5=8F=96=E6=B6=88=E4=B8=BB?= =?UTF-8?q?=E5=8A=A8=E9=87=8A=E6=94=BE=E6=98=BE=E5=AD=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- config/photo_review.py | 16 ---------------- photo_review/photo_review.py | 6 +++--- 2 files changed, 3 insertions(+), 19 deletions(-) diff --git a/config/photo_review.py b/config/photo_review.py index efba2a7..9f4865b 100644 --- a/config/photo_review.py +++ b/config/photo_review.py @@ -1,7 +1,3 @@ -from paddlenlp import Taskflow - -from config.keys import SETTLEMENT_LIST_SCHEMA, DISCHARGE_RECORD_SCHEMA, COST_LIST_SCHEMA - # 项目奔溃的重试次数 RETRY_TIME = 1 @@ -19,15 +15,3 @@ LAYOUT_ANALYSIS = False # 可使用的GPU CUDA_VISIBLE_DEVICES = "1" - -# # 基本医保结算单 -# SETTLEMENT_IE = Taskflow("information_extraction", schema=SETTLEMENT_LIST_SCHEMA, model="uie-x-base", -# task_path="config/model/settlement_list_model", layout_analysis=LAYOUT_ANALYSIS) -# -# # 出院记录 -# DISCHARGE_IE = Taskflow("information_extraction", schema=DISCHARGE_RECORD_SCHEMA, model="uie-x-base", -# task_path="config/model/discharge_record_model", layout_analysis=LAYOUT_ANALYSIS) -# -# # 费用清单 -# COST_IE = Taskflow("information_extraction", schema=COST_LIST_SCHEMA, model="uie-x-base", -# task_path="config/model/cost_list_model", layout_analysis=LAYOUT_ANALYSIS) diff --git a/photo_review/photo_review.py b/photo_review/photo_review.py index db67652..c3d1bb9 100644 --- a/photo_review/photo_review.py +++ b/photo_review/photo_review.py @@ -173,9 +173,9 @@ def information_extraction(ie, phrecs): result = merge_result(result, ie_result) - # 完成一次任务释放显存 - paddle.device.cuda.empty_cache() - logging.info("显存释放完成") + # # 完成一次任务释放显存 + # paddle.device.cuda.empty_cache() + # logging.info("显存释放完成") return result