gguf: SpecialVocab: Actually set load_merges in object

ggerganov · Aug 29, 2023 · 2ea1338 · 2ea1338
1 parent 0c620ef
commit 2ea1338
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/gguf-py/gguf/gguf.py b/gguf-py/gguf/gguf.py
@@ -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)