|
|
@@ -22,4 +22,12 @@ TRT_CFG = {
|
|
|
"SegFormer-B5": {"optimization_level": 4, "workspace_size": 1 << 32},
|
|
|
"LaTeX_OCR_rec": {"disable_ops": ["pd_op.slice"]},
|
|
|
"PP-YOLOE_seg-S": {"disable_ops": ["pd_op.slice", "pd_op.bilinear_interp"]},
|
|
|
+ "PP-FormulaNet-L": {
|
|
|
+ "disable_ops": ["pd_op.full_with_tensor"],
|
|
|
+ "workspace_size": 1 << 32,
|
|
|
+ },
|
|
|
+ "PP-FormulaNet-S": {
|
|
|
+ "disable_ops": ["pd_op.full_with_tensor"],
|
|
|
+ "workspace_size": 1 << 32,
|
|
|
+ },
|
|
|
}
|