mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-16 16:11:02 +00:00
Apply patch [skip ci]
This commit is contained in:
parent
3a3530e160
commit
e65124c56e
@ -19,7 +19,8 @@ class SentenceTransformersTokenTextSplitter(TextSplitter):
|
|||||||
super().__init__(**kwargs, chunk_overlap=chunk_overlap)
|
super().__init__(**kwargs, chunk_overlap=chunk_overlap)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
from transformers import AutoConfig, AutoTokenizer
|
from transformers import AutoConfig # type: ignore[import]
|
||||||
|
from transformers import AutoTokenizer # type: ignore[import]
|
||||||
except ImportError:
|
except ImportError:
|
||||||
msg = (
|
msg = (
|
||||||
"Could not import transformers python package. "
|
"Could not import transformers python package. "
|
||||||
@ -110,3 +111,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