Skip to content

Commit b37aeb9

Browse files
authored
Enable cache in SentenceTransformerEmbed and PdfToMarkdown. (#16)
* Enable cache for `SentenceTransformerEmbed`. * Enable cache for `PdfToMarkdown` in `pdf_embedding` example.
1 parent d583981 commit b37aeb9

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

examples/pdf_embedding/pdf_embedding.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
class PdfToMarkdown(cocoindex.op.FunctionSpec):
1212
"""Convert a PDF to markdown."""
1313

14-
@cocoindex.op.executor_class(gpu=True)
14+
@cocoindex.op.executor_class(gpu=True, cache=True, behavior_version=1)
1515
class PdfToMarkdownExecutor:
1616
"""Executor for PdfToMarkdown."""
1717

python/cocoindex/functions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ class SentenceTransformerEmbed(op.FunctionSpec):
1616
"""Run the sentence transformer"""
1717
model: str
1818

19-
@op.executor_class(gpu=True)
19+
@op.executor_class(gpu=True, cache=True, behavior_version=1)
2020
class SentenceTransformerEmbedExecutor:
2121
"""Executor for SentenceTransformerEmbed."""
2222

0 commit comments

Comments
 (0)