|
|
@@ -20,4 +20,5 @@ TRT_CFG = {
|
|
|
"SegFormer-B3": {"optimization_level": 4, "workspace_size": 1 << 32},
|
|
|
"SegFormer-B4": {"optimization_level": 4, "workspace_size": 1 << 32},
|
|
|
"SegFormer-B5": {"optimization_level": 4, "workspace_size": 1 << 32},
|
|
|
+ "LaTeX_OCR_rec": {"disable_ops": ["pd_op.slice"]},
|
|
|
}
|