CodeCow/surya
1
0
Fork 0
mirror of https://github.com/VikParuchuri/surya.git synced 2026-06-04 21:03:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Avoid truncating layout and table

Browse Source
This commit is contained in:
Vik Paruchuri 2025-08-15 11:18:31 -04:00
parent f2eecf1ad1
commit 14e7ee6ed9
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
surya/foundation/__init__.py
View File

@ -819,6 +819,11 @@ class FoundationPredictor(BasePredictor):
] or (
drop_repeated_tokens
and detect_repeat_token(predicted_tokens[p_idx])
and task_names[p_idx]
in [
TaskNames.ocr_with_boxes,
TaskNames.ocr_without_boxes,
]
)
if (
token