mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-16 08:06:14 +00:00
Apply patch [skip ci]
This commit is contained in:
parent
3aa646cb27
commit
4e70c7aee9
@ -19,7 +19,7 @@ class SentenceTransformersTokenTextSplitter(TextSplitter):
|
|||||||
super().__init__(**kwargs, chunk_overlap=chunk_overlap)
|
super().__init__(**kwargs, chunk_overlap=chunk_overlap)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
from transformers import AutoConfig, AutoTokenizer # type: ignore[import]
|
from transformers import AutoConfig, AutoTokenizer # type: ignore[attr-defined]
|
||||||
except ImportError:
|
except ImportError:
|
||||||
msg = (
|
msg = (
|
||||||
"Could not import transformers python package. "
|
"Could not import transformers python package. "
|
||||||
@ -110,3 +110,4 @@ class SentenceTransformersTokenTextSplitter(TextSplitter):
|
|||||||
truncation="do_not_truncate",
|
truncation="do_not_truncate",
|
||||||
)
|
)
|
||||||
return cast("list[int]", token_ids_with_start_and_end_token_ids)
|
return cast("list[int]", token_ids_with_start_and_end_token_ids)
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user