Skip to content

Commit f06664a

Browse files
committed
Remove base_prompt from layout injection prompt
Signed-off-by: Christoph Auer <[email protected]>
1 parent 0be5ebc commit f06664a

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
@@ -114,12 +114,13 @@ def build_prompt(self, page):
114114
layout_xml = (
115115
"<layout>" + "\n".join(layout_elements) + "</layout>"
116116
)
117-
layout_injection = f"\n{layout_xml}"
117+
layout_injection = f"{layout_xml}"
118118

119119
print(
120-
f"Layout injection prompt: {base_prompt + layout_injection}"
120+
f"Layout injection prompt: {layout_injection}"
121121
)
122-
return base_prompt + layout_injection
122+
return layout_injection
123+
#return base_prompt + layout_injection
123124

124125
return base_prompt
125126

0 commit comments

Comments
 (0)