vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

GGUFWriter compatibility fix

Browse source 
Co-authored-by: compilade <git@compilade.net>
This commit is contained in:
Christian Zhou-Zheng 2024-06-06 08:28:10 -04:00 • committed by GitHub
parent 706bd69023
commit 6a05183b97
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
gguf-py/gguf/gguf_manager.py
View file

@ -155,7 +155,7 @@ class GGUFManager(GGUFWriter):
self.state = WriterState.KV_DATA self.state = WriterState.KV_DATA
def write_tensors_to_file(self, progress: bool = False) -> None: def write_tensors_to_file(self, *, progress: bool = False) -> None:
if self.state is not WriterState.KV_DATA: if self.state is not WriterState.KV_DATA:
raise ValueError(f'Expected GGUFManager state to be KV_DATA, got {self.state}') raise ValueError(f'Expected GGUFManager state to be KV_DATA, got {self.state}')