GGUFWriter compatibility fix

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

View file

@ -155,7 +155,7 @@ class GGUFManager(GGUFWriter):
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:
raise ValueError(f'Expected GGUFManager state to be KV_DATA, got {self.state}')