Skip to content

Commit d1e51e2

Browse files
authored
modify pdf (#7738)
1 parent dca0575 commit d1e51e2

File tree

1 file changed

+2
-0
lines changed
  • pipelines/pipelines/nodes/file_converter

1 file changed

+2
-0
lines changed

pipelines/pipelines/nodes/file_converter/pdf.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -175,6 +175,8 @@ def _read_pdf(self, file_path: Path, layout: bool, process_num: int) -> List[str
175175
pdf = pypdf.PdfReader(file_path)
176176
page_length = len(pdf.pages)
177177
split_len = page_length // process_num
178+
if split_len == 0:
179+
split_len = page_length
178180
page_list = [i for i in range(0, page_length, split_len)]
179181
if page_length > page_list[-1]:
180182
page_list.append(page_length)

0 commit comments

Comments
 (0)