wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 06:40:48 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

fix legacy kwargs

Browse Source
This commit is contained in:
doctorpangloss 2025-05-17 16:10:54 -07:00
parent bbf5b9d6cf
commit a3ad9bdb1a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/text_encoders/spiece_tokenizer.py
View File

@ -10,7 +10,7 @@ class SPieceTokenizer:
def from_pretrained(path, **kwargs):
return SPieceTokenizer(path, **kwargs)
def __init__(self, tokenizer_path: bytes | str | Path, add_bos=False, add_eos=True):
def __init__(self, tokenizer_path: bytes | str | Path, add_bos=False, add_eos=True, **kwargs):
self.add_bos = add_bos
self.add_eos = add_eos
if torch.is_tensor(tokenizer_path):
@ -18,7 +18,7 @@ class SPieceTokenizer:
construction_args = {
'add_bos': self.add_bos,
'add_eos': self.add_eos
'add_eos': self.add_eos,
}
if isinstance(tokenizer_path, bytes):