Skip to content

Commit 1de6ee8

Browse files
authored
support PP-DocLayoutV2 for DCU and XPU (#4705)
1 parent 9a1154e commit 1de6ee8

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

paddlex/inference/models/doc_vlm/predictor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -481,6 +481,7 @@ def _process(item):
481481
}
482482
],
483483
return_future=True,
484+
timeout=600,
484485
**kwargs,
485486
)
486487
return future

paddlex/utils/custom_device_list.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,7 @@
5151
"RLinear",
5252
"NLinear",
5353
"PicoDet_LCNet_x2_5_face",
54+
"PP-DocLayoutV2",
5455
]
5556

5657
MLU_WHITELIST = [
@@ -215,6 +216,7 @@
215216
"PP-OCRv5_server_det",
216217
"PP-OCRv5_mobile_det",
217218
"PP-Chart2Table",
219+
"PP-DocLayoutV2",
218220
]
219221

220222
GCU_WHITELIST = [

0 commit comments

Comments
 (0)