Browse Source

Refactor:naive_merge_with_images close useless images (#9296)

### What problem does this PR solve?

naive_merge_with_images close useless images

### Type of change

- [x] Refactoring
tags/v0.20.1
Stephen Hu 2 months ago
parent
commit
0a0bfc02a0
No account linked to committer's email address
1 changed files with 4 additions and 0 deletions
  1. 4
    0
      rag/nlp/__init__.py

+ 4
- 0
rag/nlp/__init__.py View File

@@ -611,6 +611,10 @@ def naive_merge_with_images(texts, images, chunk_token_num=128, delimiter="\n。
if re.match(f"^{dels}$", sub_sec):
continue
add_chunk(sub_sec, image)
for img in images:
if isinstance(img, Image.Image):
img.close()

return cks, result_images


Loading…
Cancel
Save