gguf: SpecialVocab: Actually set load_merges in object
This commit is contained in:
parent
0c620ef63b
commit
2ea133895c
1 changed files with 1 additions and 0 deletions
|
@ -761,6 +761,7 @@ class SpecialVocab:
|
|||
|
||||
def __init__(self, path: Path, load_merges: bool = False, special_token_types: Optional[Tuple[str, ...]] = None):
|
||||
self.special_token_ids = {}
|
||||
self.load_merges = load_merges
|
||||
if special_token_types is not None:
|
||||
self.special_token_types = special_token_types
|
||||
self.load(path)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue