Skip to content

Commit abdc907

Browse files
committed
Pipeline fix
1 parent e691ea1 commit abdc907

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

olmocr/pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -238,7 +238,7 @@ async def process_page(args, worker_id: int, pdf_orig_path: str, pdf_local_path:
238238

239239
# Enable guided decoding regex if needed
240240
if args.guided_decoding:
241-
query["guided_regex"] = r"---\nprimary_language: .{1,20}\nis_rotation_valid: (?:True|False|true|false)\nrotation_correction:(?:0|90|180|270)\nis_table: (?:True|False|true|false)\nis_diagram: (?:True|False|true|false)\n---\n[\s\S]*"
241+
query["guided_regex"] = r"---\nprimary_language: .{1,20}\nis_rotation_valid: (?:True|False|true|false)\nrotation_correction: (?:0|90|180|270)\nis_table: (?:True|False|true|false)\nis_diagram: (?:True|False|true|false)\n---\n[\s\S]*"
242242

243243
logger.info(f"Built page query for {pdf_orig_path}-{page_num}")
244244

0 commit comments

Comments
 (0)