mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-15 15:46:47 +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)
|
||||
|
||||
try:
|
||||
from transformers import AutoConfig, AutoTokenizer
|
||||
from transformers import AutoConfig # type: ignore[import]
|
||||
from transformers import AutoTokenizer # type: ignore[import]
|
||||
except ImportError:
|
||||
msg = (
|
||||
"Could not import transformers python package. "
|
||||
@ -110,3 +111,4 @@ class SentenceTransformersTokenTextSplitter(TextSplitter):
|
||||
truncation="do_not_truncate",
|
||||
)
|
||||
return cast("list[int]", token_ids_with_start_and_end_token_ids)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user