wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 23:00:51 +08:00
Code Issues Actions 10 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): def from_pretrained(path, **kwargs):
return SPieceTokenizer(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_bos = add_bos
self.add_eos = add_eos self.add_eos = add_eos
if torch.is_tensor(tokenizer_path): if torch.is_tensor(tokenizer_path):
@ -18,7 +18,7 @@ class SPieceTokenizer:
construction_args = { construction_args = {
'add_bos': self.add_bos, 'add_bos': self.add_bos,
'add_eos': self.add_eos 'add_eos': self.add_eos,
} }
if isinstance(tokenizer_path, bytes): if isinstance(tokenizer_path, bytes):