Skip to content

Commit f248bbd

Browse files
committed
Reinstate custom prompt
Signed-off-by: Christoph Auer <[email protected]>
1 parent 77107fe commit f248bbd

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

docling/experimental/pipeline/threaded_layout_vlm_pipeline.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -116,9 +116,10 @@ def build_prompt(self, page):
116116
)
117117
layout_injection = f"{layout_xml}"
118118

119-
print(f"Layout injection prompt: {layout_injection}")
120-
return layout_injection
121-
# return base_prompt + layout_injection
119+
custom_prompt = base_prompt + layout_injection
120+
print(f"Layout injection prompt: {custom_prompt}")
121+
122+
return custom_prompt
122123

123124
return base_prompt
124125

0 commit comments

Comments
 (0)