测试显存占用

test.py

@@ -0,0 +1,17 @@
+from paddlenlp import Taskflow
+
+SCHEMA = ["患者姓名", "入院日期", "出院日期", "费用总额", "个人现金支付", "个人账户支付", "自费金额", "医保类型",
+          "住院号", "医保结算单号码"]
+
+IE = Taskflow("information_extraction", schema=SCHEMA, model="uie-x-base",
+              task_path="config/model/settlement_list_model", layout_analysis=True)
+
+
+def main():
+    i = 0
+    try:
+        while 1:
+            IE({"doc": "visual_model_test/img/PH20240511000638_1_094306_1.jpg"})
+            i += 1
+    except Exception:
+        print(i)